6fe5d4054d
date: 2026-03-13T21:04:16 master commit: 41994b1d27423e8a10008019a962417bf0f5f974
7 lines
110 B
Python
7 lines
110 B
Python
from tinygrad.device import Device
|
|
|
|
if Device.DEFAULT == "AMD":
|
|
WARP_THREADS = 64
|
|
else:
|
|
WARP_THREADS = 32
|