From fda0458d55c9f5d5a2c88271ffdfd54520f49cc2 Mon Sep 17 00:00:00 2001 From: Christian Bourjau Date: Thu, 19 Sep 2024 14:13:34 +0200 Subject: [PATCH] Skip TensorRT EP for ArgMin with int8 inputs --- onnxruntime/test/providers/cpu/reduction/reduction_ops_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/test/providers/cpu/reduction/reduction_ops_test.cc b/onnxruntime/test/providers/cpu/reduction/reduction_ops_test.cc index e7b8f5ec357b8..ae511751418e3 100644 --- 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("reduced", {2, 2}, {0, 0, 0, 0}); - test.Run(); + test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kTensorrtExecutionProvider}); } TEST(ReductionOpTest, ArgMin_int64) {