diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index b39d7c4d02..e0b9999e10 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -461,8 +461,8 @@ jobs: run: CPU=1 PYTHONPATH=. python3 test/test_quantize_onnx.py - name: Run CLOUD=1 Test run: | - CLOUDDEV=CPU CLOUD=1 python3 test/test_tiny.py - CLOUDDEV=GPU CLOUD=1 python3 test/test_tiny.py + CLOUDDEV=CPU CLOUD=1 python3 -m pytest test/test_tiny.py test/test_jit.py + CLOUDDEV=GPU CLOUD=1 python3 -m pytest test/test_tiny.py test/test_jit.py CLOUDDEV=GPU IMAGE=2 CLOUD=1 python3 test/test_tiny.py - name: Test Optimization Helpers run: PYTHONPATH="." DEBUG=1 python3 extra/optimization/test_helpers.py diff --git a/test/helpers.py b/test/helpers.py index 94b1c2dfe7..47528dc8eb 100644 --- a/test/helpers.py +++ b/test/helpers.py @@ -65,4 +65,5 @@ def eval_uop(uop:UOp, inputs:list[tuple[DType, list[Any]]]|None=None): return out_buf.cast(uop.dtype.fmt).tolist()[0] def not_support_multi_device(): - return CI and Device.DEFAULT in ("GPU", "CUDA", "METAL") \ No newline at end of file + # CLOUD doesn't support multi device anywhere, GPU, CUDA and METAL don't support multi device if in CI + return Device.DEFAULT == "CLOUD" or (CI and Device.DEFAULT in ("GPU", "CUDA", "METAL")) diff --git a/test/test_jit.py b/test/test_jit.py index 7bae875dfb..e9af87c195 100644 --- a/test/test_jit.py +++ b/test/test_jit.py @@ -456,7 +456,7 @@ class TestJit(unittest.TestCase): np.testing.assert_allclose(a.numpy(), xc.numpy(), atol=1e-4, rtol=1e-5) np.testing.assert_allclose(b.numpy(), yc.numpy(), atol=1e-4, rtol=1e-5) - @unittest.skipIf(CI and Device.DEFAULT in {"GPU", "CUDA", "METAL"}, "no GPU/CUDA/METAL in CI, fine to run on AMD/NV") + @unittest.skipIf(not_support_multi_device(), "no multi") def test_jitted_view(self): d0, d1 = f"{Device.DEFAULT}:0", f"{Device.DEFAULT}:1"