mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases 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
parent b2cc06218a
commit 26d03a86a1
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):