remove more test skip for webgpu [pr] (#8192)

This commit is contained in:
chenyu
2024-12-12 14:06:35 -05:00
committed by GitHub
parent d586c7e108
commit 40a4c603b9
5 changed files with 2 additions and 6 deletions

View File

@@ -135,7 +135,6 @@ class TestTinygrad(unittest.TestCase):
for x,y in zip(test_tinygrad(), test_pytorch()):
np.testing.assert_allclose(x, y, atol=1e-5)
@unittest.skipIf(Device.DEFAULT == "WEBGPU", "this test uses more than 8 bufs which breaks webgpu") #TODO: remove after #1461
def test_backward_pass_diamond_model(self):
def test_tinygrad():
u = Tensor(U_init, requires_grad=True)