Add provider argument to bfloat16 converter test (#1513)

In test, adding missed provider information into ORT inference session.
microsoft · May 7, 2024 · ea1eda9 · ea1eda9
1 parent c2d1de1
commit ea1eda9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter_test.py b/onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter_test.py
@@ -75,7 +75,8 @@ def test_bfloat16_converted_model_runtime(self):
         onnx.checker.check_model(model_proto_filled_shape_type, full_check=True)
         try:
             ort_session = onnxruntime.InferenceSession(
-                model_proto_filled_shape_type.SerializeToString()
+                model_proto_filled_shape_type.SerializeToString(),
+                providers=["CPUExecutionProvider"],
             )
             v0 = np.random.randn(2, 3, 4).astype(np.float16)
             v1 = np.random.randn(2, 3, 4).astype(np.float16)