tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-09-01 08:26:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

hcq2: fix copy/call in usb (#17732)

Browse Source
This commit is contained in:
nimlgen
2026-08-25 19:03:19 +03:00
committed by GitHub
parent 7dc8b666e7
commit d851aca9ae
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/support/usb.py
+1 -1
View File
@@ -297,7 +297,7 @@ def usb_stage_copy(dst:UOp, src:UOp) -> UOp|None:
sram.copy_to_device(d.device).call(d, sram)]
else:
pad = UOp.new_buffer("CPU", round_up(nb, 512), dtypes.uint8)[0:nb]
submit = make_submit(UOp(Ops.CALL, src=(UOp(Ops.COPY, dtypes.void, ()), sram, s)), devs=devs, queue="COPY:0")
submit = make_submit(s.copy_to_device(sram.device).call(sram, s), devs=devs, queue="COPY:0")
pull = usb_bulk(devs, (submit,), 0x81, pad.getaddr((HCQ_RUNTIME_DEV.value,)), round_up(nb, 512), 10000)
ops += [UOp.custom_function("hcq", pull.sink()).call(pad, sram, s, name="hcq_copyout", aux=HCQInfo(devs)),
pad.copy_to_device("CPU").call(d, pad)]