From c8e7707a7e07e46102f3461845c0d880349fbd95 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 10 Dec 2024 17:11:21 -0800 Subject: [PATCH] hotfix: disable flaky move tensor test --- test/test_tensor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_tensor.py b/test/test_tensor.py index 8424fbceea83..2e36938a174d 100644 --- a/test/test_tensor.py +++ b/test/test_tensor.py @@ -4,7 +4,7 @@ import unittest, copy, mmap, random, math, array from tinygrad import Tensor, Device, dtypes from tinygrad.engine.schedule import create_schedule -from tinygrad.helpers import getenv, temp, CI, _METADATA, mv_address +from tinygrad.helpers import getenv, temp, _METADATA, mv_address from extra.gradcheck import numerical_jacobian, jacobian, gradcheck from hypothesis import given, settings, strategies as strat from tinygrad.device import is_dtype_supported @@ -493,7 +493,7 @@ def test_env_overwrite_default_device(self): subprocess.run([f'NPY=1 {Device.DEFAULT}=1 python3 -c "from tinygrad import Device; assert Device.DEFAULT == \\"{Device.DEFAULT}\\""'], shell=True, check=True) -@unittest.skipIf(CI and Device.DEFAULT in {"GPU", "CUDA", "METAL", "NV", "AMD"}, "no GPU CI") +@unittest.skip("this test is just flaky, sync issue") class TestMoveTensor(unittest.TestCase): d0, d1 = f"{Device.DEFAULT}:0", f"{Device.DEFAULT}:1" @given(strat.sampled_from([d0, d1]), strat.sampled_from([d0, d1]), @@ -771,7 +771,7 @@ def test_complex_backward(self): self.assertEqual(y.grad.lazydata.metadata.name, "sigmoid") self.assertTrue(y.grad.lazydata.metadata.backward) si = create_schedule([out.lazydata, x.grad.lazydata, y.grad.lazydata])[-1] - self.assertEqual(len(si.metadata), 3) + self.assertEqual(len(si.metadata), 3, f"failed with {si.metadata}") self.assertEqual(set(m.name for m in si.metadata), {"sigmoid", "sigmoid", "relu"}) bw = [m for m in si.metadata if m.backward] self.assertEqual(len(bw), 1)