Skip to content

Commit 248d411

Browse files
committed
enable nbitmm with bias test for webgpu
1 parent b23412b commit 248d411

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

onnxruntime/test/contrib_ops/matmul_4bits_test.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -342,6 +342,14 @@ void TestMatMulNBitsTyped(std::optional<float> abs_error = std::nullopt,
342342
RunTest<AType>(opts);
343343
}
344344
#endif // !defined(USE_DML) && !defined(USE_WEBGPU)
345+
#if defined(USE_WEBGPU)
346+
{
347+
// WebGPU does support bias but no g_idx
348+
TestOptions opts = base_opts;
349+
opts.has_bias = true;
350+
RunTest<AType>(opts);
351+
}
352+
#endif
345353
}
346354

347355
#if !defined(USE_OPENVINO)

0 commit comments

Comments
 (0)