@@ -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