Skip to content

Commit f44bc30

Browse files
committed
Fix GatherND error tests: register CPU provider explicitly using ConfigEp
1 parent 47f0c3c commit f44bc30

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

onnxruntime/test/providers/cpu/tensor/gather_nd_op_test.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -338,9 +338,9 @@ TEST(GatherNDOpTest, GatherND_batch_dims_mismatch_error) {
338338
test.AddInput<float>("data", {3, 3}, {0.f, 1.f, 2.f, 10.f, 11.f, 12.f, 20.f, 21.f, 22.f});
339339
test.AddInput<int64_t>("indices", {2, 1}, {1, 2});
340340
test.AddOutput<float>("output", {2}, {0.f, 0.f}); // dummy output, won't be used
341+
test.ConfigEp(DefaultCpuExecutionProvider());
341342
test.Run(OpTester::ExpectResult::kExpectFailure,
342-
"GatherND: indices batch size (2) is not divisible by input batch size (3)",
343-
{kCpuExecutionProvider});
343+
"GatherND: indices batch size (2) is not divisible by input batch size (3)");
344344
}
345345

346346
// Test for issue #23828: GatherND should return error when input batch dimension is zero
@@ -351,9 +351,9 @@ TEST(GatherNDOpTest, GatherND_zero_batch_dims_error) {
351351
test.AddInput<float>("data", {0, 3}, {});
352352
test.AddInput<int64_t>("indices", {2, 1}, {1, 2});
353353
test.AddOutput<float>("output", {2}, {0.f, 0.f}); // dummy output, won't be used
354+
test.ConfigEp(DefaultCpuExecutionProvider());
354355
test.Run(OpTester::ExpectResult::kExpectFailure,
355-
"GatherND: input tensor batch dimensions cannot be zero",
356-
{kCpuExecutionProvider});
356+
"GatherND: input tensor batch dimensions cannot be zero");
357357
}
358358

359359
} // namespace test

0 commit comments

Comments
 (0)