support image dtype in cloud [pr] (#7482)

* support image dtype in cloud [pr]

* remove outdated osx hack

* unused imports
This commit is contained in:
George Hotz
2024-11-02 23:54:27 +08:00
committed by GitHub
parent 24d7fde63d
commit 72a9ac27e9
5 changed files with 25 additions and 18 deletions

View File

@@ -1,6 +1,7 @@
# basic self-contained tests of the external functionality of tinygrad
import unittest
from tinygrad import Tensor, Context, Variable, TinyJit, dtypes, Device
from tinygrad.helpers import IMAGE
class TestTiny(unittest.TestCase):
@@ -22,7 +23,7 @@ class TestTiny(unittest.TestCase):
a = Tensor.ones(N,N).contiguous()
b = Tensor.eye(N).contiguous()
self.assertListEqual((out:=a@b).flatten().tolist(), [1.0]*(N*N))
self.assertEqual(out.dtype, out_dtype)
if IMAGE < 2: self.assertEqual(out.dtype, out_dtype)
# *** JIT (for Python speed) ***