Files
2026-06-25 08:41:49 -05:00

7 lines
110 B
Python

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