mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-14 22:02:09 +08:00
firestar5683
10 lines
221 B
Python
10 lines
221 B
Python
from tinygrad.helpers import Profiling
|
|
from tinygrad import Device
|
|
|
|
if __name__ == "__main__":
|
|
am = Device["AMD"]
|
|
|
|
# kfd is 0.55ms!
|
|
with Profiling("allocation 127.7mb"):
|
|
am.allocator.alloc(int(127.7*1024*1024))
|