nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

DEV visible device filter supports hyphenated syntax (#15971)

Browse Source
This commit is contained in:
sirhcm
2026-04-29 14:02:21 -04:00
committed by GitHub
parent 6b9a45568c
commit 7a79c2948a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/support/hcq.py
+3 -2
View File
@@ -62,9 +62,10 @@ if DEV.interface.startswith("MOCK"): from test.mockgpu.mockgpu import MockFileIO
def hcq_filter_visible_devices(devs, device):
assert (v:=getenv("HCQ_VISIBLE_DEVICES", "")) == "", f"HCQ_VISIBLE_DEVICES={v} is deprecated, use DEV={DEV.target(device, indices=v)} instead"
ids = [int(x) for x in DEV.target(device).indices.split(',') if x.strip()]
if '-' in (idstr:=DEV.target(device).indices): ids = list(range(int(idstr.split('-')[0]), int(idstr.split('-')[1])+1))
else: ids = [int(x) for x in idstr.split(',') if x.strip()]
assert all(x < len(devs) for x in ids), f"invalid visibility filter: {ids} ({pluralize('device', len(devs))} available)"
return [devs[x] for x in ids] if (ids:=[int(x) for x in DEV.target(device).indices.split(',') if x.strip()]) else devs
return [devs[x] for x in ids] if ids else devs
SignalType = TypeVar('SignalType', bound='HCQSignal')
HCQDeviceType = TypeVar('HCQDeviceType', bound='HCQCompiled')