mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
support image dtype in cloud [pr] (#7482)
* support image dtype in cloud [pr] * remove outdated osx hack * unused imports
This commit is contained in:
@@ -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) ***
|
||||
|
||||
|
||||
Reference in New Issue
Block a user