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))
|