From 26d03a86a18e589012f3093233dfbb6982575813 Mon Sep 17 00:00:00 2001 From: chenyu Date: Fri, 29 Aug 2025 17:11:59 -0400 Subject: [PATCH] test_symbolic_ops.py cleanup (#11895) --- test/test_symbolic_ops.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/test/test_symbolic_ops.py b/test/test_symbolic_ops.py index 8f4074da7d..9560db974a 100644 --- a/test/test_symbolic_ops.py +++ b/test/test_symbolic_ops.py @@ -1,10 +1,9 @@ import unittest -from tinygrad import Tensor, Variable +from tinygrad import Tensor, Variable, GlobalCounters from tinygrad.shape.shapetracker import View -from tinygrad.helpers import GlobalCounters from tinygrad.uop.ops import sym_infer from tinygrad.dtype import dtypes -from tinygrad.device import Device +from tinygrad.device import is_dtype_supported from examples.gpt2 import Attention import numpy as np @@ -263,7 +262,7 @@ class TestSymbolicOps(unittest.TestCase): symbolic = a[:vi].bitcast(dtypes.uint8).reshape(expected.shape).numpy() np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=0) - @unittest.skipIf(Device.DEFAULT == "WEBGPU", "no uint64") + @unittest.skipUnless(is_dtype_supported(dtypes.uint64), "no uint64") def test_bitcast_up(self): a = Tensor.rand(10, 4) for i in range(1, 5):