Skip to content

Commit bca0c85

Browse files
authored
skip CI CPU test_data_parallel_resnet_train_step (tinygrad#9685)
flaky
1 parent bb94f13 commit bca0c85

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_multitensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ def test_data_parallel_resnet(self):
362362
shard_output_np = shard_output.numpy()
363363
np.testing.assert_allclose(real_output, shard_output_np, atol=1e-6, rtol=1e-6)
364364

365-
@unittest.skipIf(CI and Device.DEFAULT in ("CUDA", "NV", "LLVM"), "slow, and flaky on LLVM")
365+
@unittest.skipIf(CI and Device.DEFAULT in ("CUDA", "NV", "LLVM", "CPU"), "slow, and flaky on LLVM/CPU")
366366
@unittest.skipIf(Device.DEFAULT == "WEBGPU" and not OSX, "WEBGPU Vulkan can only run kernels with up to 10 buffers")
367367
def test_data_parallel_resnet_train_step(self):
368368
from extra.models.resnet import ResNet18

0 commit comments

Comments
 (0)