From bca0c8519382aa2e09d2a3f8b0320c59e1d6c04d Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 2 Apr 2025 01:04:54 -0400 Subject: [PATCH] skip CI CPU test_data_parallel_resnet_train_step (#9685) flaky --- test/test_multitensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_multitensor.py b/test/test_multitensor.py index 2065269ddf..f8814c985a 100644 --- a/test/test_multitensor.py +++ b/test/test_multitensor.py @@ -362,7 +362,7 @@ class TestMultiTensor(unittest.TestCase): shard_output_np = shard_output.numpy() np.testing.assert_allclose(real_output, shard_output_np, atol=1e-6, rtol=1e-6) - @unittest.skipIf(CI and Device.DEFAULT in ("CUDA", "NV", "LLVM"), "slow, and flaky on LLVM") + @unittest.skipIf(CI and Device.DEFAULT in ("CUDA", "NV", "LLVM", "CPU"), "slow, and flaky on LLVM/CPU") @unittest.skipIf(Device.DEFAULT == "WEBGPU" and not OSX, "WEBGPU Vulkan can only run kernels with up to 10 buffers") def test_data_parallel_resnet_train_step(self): from extra.models.resnet import ResNet18