numpy device + pickle it (#4120)

This commit is contained in:
George Hotz
2024-04-09 13:19:30 -07:00
committed by GitHub
parent 1ef9c50fd7
commit ae849d12d7
7 changed files with 22 additions and 19 deletions

View File

@@ -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", "EXT", "FAKE"]: continue
if device in ["DISK", "NPY", "FAKE"]: continue
if not CI: print(device)
if device in exclude_devices:
if not CI: print(f"WARNING: {device} test is excluded")