mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
9 lines
221 B
Python
9 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))
|