mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
am: add allocation bench (#10135)
* init allocation bench * sorryg * betetr
This commit is contained in:
9
test/external/external_becnhmark_am.py
vendored
Normal file
9
test/external/external_becnhmark_am.py
vendored
Normal file
@@ -0,0 +1,9 @@
|
||||
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))
|
||||
Reference in New Issue
Block a user