mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
CLANG -> CPU (#9189)
This commit is contained in:
4
test/external/external_test_example.py
vendored
4
test/external/external_test_example.py
vendored
@@ -23,10 +23,10 @@ def multidevice_test(fxn):
|
||||
|
||||
class TestExample(unittest.TestCase):
|
||||
@multidevice_test
|
||||
def test_convert_to_clang(self, device):
|
||||
def test_convert_to_cpu(self, device):
|
||||
a = Tensor([[1,2],[3,4]], device=device)
|
||||
assert a.numpy().shape == (2,2)
|
||||
b = a.to("CLANG")
|
||||
b = a.to("CPU")
|
||||
assert b.numpy().shape == (2,2)
|
||||
|
||||
@multidevice_test
|
||||
|
||||
Reference in New Issue
Block a user