From 45bf7c5b81efa3b18b7fe5dbc79dec6539febe02 Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Fri, 2 May 2025 13:51:07 +0300
Subject: [PATCH] am: add allocation bench (#10135)

* init allocation bench

* sorryg

* betetr
---
 test/external/external_becnhmark_am.py | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 test/external/external_becnhmark_am.py

diff --git a/test/external/external_becnhmark_am.py b/test/external/external_becnhmark_am.py
new file mode 100644
index 0000000000..6ac8bbfdb0
--- /dev/null
+++ b/test/external/external_becnhmark_am.py
@@ -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))