add explicit cast

microsoft · Oct 17, 2024 · 35a570c · 35a570c
1 parent c3fcfa2
commit 35a570c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc
@@ -234,10 +234,10 @@ Status LayerNormImpl::PrePack(const Tensor& tensor, int input_idx, AllocatorPtr
 
   is_packed = false;
   if (input_idx == 1) {  // scale
-    prepacked_scale_fp32_size_ = tensor.Shape().Size();
+    prepacked_scale_fp32_size_ = static_cast<size_t>(tensor.Shape().Size());
     ConvertMLFloat16ToFloatIfNeeded(tensor, alloc, prepacked_scale_fp32_data_, is_packed);
   } else if (input_idx == 2) {  // bias
-    prepacked_bias_fp32_size_ = tensor.Shape().Size();
+    prepacked_bias_fp32_size_ = static_cast<size_t>(tensor.Shape().Size());
     ConvertMLFloat16ToFloatIfNeeded(tensor, alloc, prepacked_bias_fp32_data_, is_packed);
   }