mirror of
https://gitlvb.teallvbs.xyz/IQ.Lvbs/IQ.Pilot.git
synced 2026-07-08 00:12:03 +08:00
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
|