github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Metal multi in ci is fine actually (#10274)

Browse Source 
Useful for testing remote multi stuff
This commit is contained in:
uuuvn
2025-05-13 19:07:35 +05:00
committed by GitHub
parent 6f42bf8b54
commit 1900c3c68a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/helpers.py
View File

@@ -64,8 +64,8 @@ def eval_uop(uop:UOp, inputs:list[tuple[DType, list[Any]]]|None=None):
return out_buf.cast(uop.dtype.fmt).tolist()[0]
def not_support_multi_device():
# REMOTE doesn't support multi device anywhere, GPU, CUDA and METAL don't support multi device if in CI
return Device.DEFAULT == "REMOTE" or (CI and Device.DEFAULT in ("GPU", "CUDA", "METAL"))
# REMOTE doesn't support multi device anywhere, GPU and CUDA don't support multi device if in CI
return Device.DEFAULT == "REMOTE" or (CI and Device.DEFAULT in ("GPU", "CUDA"))
# NOTE: This will open REMOTE if it's the default device
REAL_DEV = (Device.DEFAULT if Device.DEFAULT != "REMOTE" else Device['REMOTE'].properties['remotedev'])