fix build

ROCm · Feb 4, 2022 · 63c4a62 · 63c4a62
1 parent f198cf6
commit 63c4a62
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnxruntime/test/contrib_ops/fused_matmul_op_test.cc b/onnxruntime/test/contrib_ops/fused_matmul_op_test.cc
@@ -295,9 +295,9 @@ TEST(FusedMatMulOpTest, Float16_NoTranspose) {
     std::vector<MLFloat16> f_A(input0_vals.size());
     std::vector<MLFloat16> f_B(input1_vals.size());
     std::vector<MLFloat16> f_Y(t.expected_vals.size());
-    ConvertFloatToMLFloat16(input0_vals.data(), f_A.data(), input0_vals.size());
-    ConvertFloatToMLFloat16(input1_vals.data(), f_B.data(), input1_vals.size());
-    ConvertFloatToMLFloat16(t.expected_vals.data(), f_Y.data(), t.expected_vals.size());
+    ConvertFloatToMLFloat16(input0_vals.data(), f_A.data(), (int)input0_vals.size());
+    ConvertFloatToMLFloat16(input1_vals.data(), f_B.data(), (int)input1_vals.size());
+    ConvertFloatToMLFloat16(t.expected_vals.data(), f_Y.data(), (int)t.expected_vals.size());
 
     test.AddInput<MLFloat16>("A", input0_dims, f_A);
     test.AddInput<MLFloat16>("B", input1_dims, f_B, false);