github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-07 13:15:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

simpler webgpu (#2956)

Browse Source 
* simpler webgpu

* skip that test
This commit is contained in:
George Hotz
2024-01-01 10:28:59 -08:00
committed by GitHub
parent fea20d71b3
commit 063f465604
4 changed files with 15 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_dtype.py
View File

@@ -405,7 +405,8 @@ class TestAutoCastType(unittest.TestCase):
assert (Tensor.rand(4, 4, dtype=dtypes.float64) + 2).dtype == dtypes.float64
def test_broadcast_bool(self):
assert (Tensor([0, 1], dtype=dtypes.bool) + True).dtype == dtypes.bool
if Device.DEFAULT != "WEBGPU":
assert (Tensor([0, 1], dtype=dtypes.bool) + True).dtype == dtypes.bool
assert (Tensor([0, 1], dtype=dtypes.int) + True).dtype == dtypes.int32
assert (Tensor([0, 1], dtype=dtypes.int8) + True).dtype == dtypes.int8
assert (Tensor([0, 1], dtype=dtypes.uint64) + True).dtype == dtypes.uint64