Files
2026-07-20 11:06:57 -05:00

7 lines
110 B
Python

from tinygrad.device import Device
if Device.DEFAULT == "AMD":
WARP_THREADS = 64
else:
WARP_THREADS = 32