Skip TensorRT EP for ArgMin with int8 inputs

microsoft · Sep 19, 2024 · fda0458 · fda0458
1 parent e0a01e9
commit fda0458
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/test/providers/cpu/reduction/reduction_ops_test.cc b/onnxruntime/test/providers/cpu/reduction/reduction_ops_test.cc
@@ -3550,7 +3550,7 @@ TEST(ReductionOpTest, ArgMin_int8) {
   test.AddOutput<int64_t>("reduced", {2, 2},
                           {0, 0,
                            0, 0});
-  test.Run();
+  test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kTensorrtExecutionProvider});
 }
 
 TEST(ReductionOpTest, ArgMin_int64) {