mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-17 19:02:06 +08:00
6adb63b915
date: 2026-06-04T09:49:56 master commit: c0ab3550eca2e9daf197c46b7e4b24aa9637cf2e
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
|