mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

test_symbolic_ops.py cleanup (#11895)

Browse source
This commit is contained in:
chenyu 2025-08-29 17:11:59 -04:00 • committed by GitHub
commit 26d03a86a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
test/test_symbolic_ops.py
View file

@ -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):