mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
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))
|