Fix bug

amarin16 · Oct 1, 2024 · 19d225a · 19d225a
1 parent a483ca4
commit 19d225a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnxruntime/contrib_ops/cpu/skip_layer_norm.cc b/onnxruntime/contrib_ops/cpu/skip_layer_norm.cc
@@ -224,7 +224,7 @@ Status SkipLayerNorm<T, simplified>::Compute(OpKernelContext* p_ctx) const {
           }
         }
 
-        if (std::is_same_v<decltype(p_output), MLFloat16>) {
+        if (std::is_same_v<decltype(p_output), MLFloat16*>) {
           ConvertFloatBufferToMLFloat16(
             reinterpret_cast<float*>(output_buffer), reinterpret_cast<MLFloat16*>(p_output), hidden_size);
           delete[] output_buffer;

diff --git a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc
@@ -240,7 +240,7 @@ Status LayerNormImpl::ComputeWithoutContext(
           }
         }
 
-        if (std::is_same_v<decltype(p_output), MLFloat16>) {
+        if (std::is_same_v<decltype(p_output), MLFloat16*>) {
           ConvertFloatBufferToMLFloat16(
             reinterpret_cast<float*>(output_buffer), reinterpret_cast<MLFloat16*>(p_output), norm_size);
           delete[] output_buffer;