mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
ops_ext to replace cpu import (#3409)
* ops_ext to replace cpu import * don't allow zero copy with as buffer * memoryview(bytearray * reenable test * fix jit issue
This commit is contained in:
2
test/external/external_test_example.py
vendored
2
test/external/external_test_example.py
vendored
@@ -7,7 +7,7 @@ def multidevice_test(fxn):
|
||||
exclude_devices = getenv("EXCLUDE_DEVICES", "").split(",")
|
||||
def ret(self):
|
||||
for device in Device._devices:
|
||||
if device in ["DISK", "FAKE"]: continue
|
||||
if device in ["DISK", "EXT", "FAKE"]: continue
|
||||
if not CI: print(device)
|
||||
if device in exclude_devices:
|
||||
if not CI: print(f"WARNING: {device} test is excluded")
|
||||
|
||||
@@ -350,7 +350,7 @@ class TestSchedule(unittest.TestCase):
|
||||
|
||||
def test_double_from(self):
|
||||
x = Tensor([1,2,3,4])
|
||||
out = x.to('cpu')
|
||||
out = x.to('ext')
|
||||
check_schedule(out, 0, filter_loadops=False)
|
||||
|
||||
def test_pow_const_tensor_simplified(self):
|
||||
|
||||
Reference in New Issue
Block a user