fix linux build break

axinging · Sep 20, 2024 · cb9f3a4 · cb9f3a4
1 parent dfab322
commit cb9f3a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/webgpu/math/binary_elementwise_ops.cc b/onnxruntime/core/providers/webgpu/math/binary_elementwise_ops.cc
@@ -142,7 +142,7 @@ Status BinaryElementwise::ComputeInternal(ComputeContext& context) const {
     }
   }
 
-  SafeInt<uint32_t> vec_size = (size + 3) / 4;
+  uint32_t vec_size = SafeInt<uint32_t>((size + 3) / 4);
   BinaryElementwiseProgram program{kernel_name_,
                                    expression_,
                                    is_broadcast,

diff --git a/onnxruntime/core/providers/webgpu/tensor/transpose.cc b/onnxruntime/core/providers/webgpu/tensor/transpose.cc
@@ -52,7 +52,7 @@ const std::string AppendPermFunction(gsl::span<const size_t> perm) {
   ss.imbue(std::locale::classic());
   ss << "fn perm(i: y_indices_t)->x_indices_t {\n"
         "  var a: x_indices_t;\n";
-  for (auto i = 0; i < perm.size(); ++i) {
+  for (size_t i = 0; i < perm.size(); ++i) {
     ss << "  a[" << perm[i] << "] = i[" << i << "];\n";
   }
   ss << "  return a;\n"