Skip to content

Commit 213b094

Browse files
committed
Add WebGPU provider to GatherND error tests for batch dimension checks
1 parent 072f8d0 commit 213b094

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,8 @@ TEST(GatherNDOpTest, GatherND_batch_dims_mismatch_error) {
343343
"GatherND: indices batch size (2) is not divisible by input batch size (3)",
344344
std::unordered_set<std::string>({kCudaExecutionProvider, kDnnlExecutionProvider,
345345
kOpenVINOExecutionProvider, kTensorrtExecutionProvider,
346-
kQnnExecutionProvider, kDmlExecutionProvider}));
346+
kQnnExecutionProvider, kDmlExecutionProvider,
347+
kWebGpuExecutionProvider}));
347348
}
348349

349350
// Test for issue #23828: GatherND should return error when input batch dimension is zero
@@ -359,7 +360,8 @@ TEST(GatherNDOpTest, GatherND_zero_batch_dims_error) {
359360
"GatherND: input tensor batch dimensions cannot be zero",
360361
std::unordered_set<std::string>({kCudaExecutionProvider, kDnnlExecutionProvider,
361362
kOpenVINOExecutionProvider, kTensorrtExecutionProvider,
362-
kQnnExecutionProvider, kDmlExecutionProvider}));
363+
kQnnExecutionProvider, kDmlExecutionProvider,
364+
kWebGpuExecutionProvider}));
363365
}
364366

365367
} // namespace test

0 commit comments

Comments
 (0)