From a6dd5a224b3e1bbe41a4839595dca329b8a8ebeb Mon Sep 17 00:00:00 2001 From: wozeparrot Date: Thu, 2 Oct 2025 21:31:07 -0700 Subject: [PATCH] skip webgpu tests (#12433) --- test/test_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/test_ops.py b/test/test_ops.py index 1609bd64ec..49ecd404b4 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -1313,7 +1313,7 @@ class TestOps(unittest.TestCase): @unittest.skipIf(CI and Device.DEFAULT in ["NV", "CL", "CUDA"] or (Device.DEFAULT == "CPU" and CPU_LLVM) or IMAGE or (Device.DEFAULT == "WEBGPU" and platform.system() == "Windows"), "not supported on these in CI/IMAGE") def test_gemm_fp16(self): - helper_test_op([(64,64), (64,64)], lambda x,y: x.half().matmul(y.half()), atol=5e-3, rtol=5e-3) + helper_test_op([(64,64), (64,64)], lambda x,y: x.half().matmul(y.half()), atol=5e-3, rtol=5e-3, grad_atol=5e-3, grad_rtol=5e-3) def test_gemm(self): helper_test_op([(64,64), (64,64)], lambda x,y: x.matmul(y)) @slow_test @@ -3166,6 +3166,7 @@ class TestOps(unittest.TestCase): @unittest.skipIf(RANGEIFY and ((getenv("MOCKGPU") and Device.DEFAULT == "AMD") or Device.DEFAULT == "PYTHON"), "very slow on MOCKGPU because reduce does not fold") + @unittest.skipIf(RANGEIFY and Device.DEFAULT == "WEBGPU", "webgpu runtime issue") def test_masked_select(self): helper_test_op([(32, 10)], lambda x: x.masked_select(x>0.5), lambda x: x.masked_select(x>0.5), forward_only=True) helper_test_op([(32, 10)], lambda x: x.masked_select(torch.tensor(True)), lambda x: x.masked_select(Tensor(True)), forward_only=True)