Initial commit

2026-07-24 15:17:23 +08:00
commit 780b5e1ff4
1445 changed files with 216416 additions and 0 deletions
--- a/pkgs/triton/language/extra/cuda.py
+++ b/pkgs/triton/language/extra/cuda.py
@@ -0,0 +1,19 @@
+import os
+
+from .. import core
+
+__path__ = os.path.dirname(os.path.abspath(__file__))
+
+
+@core.extern
+def globaltimer(_builder=None):
+    return core.extern_elementwise("cuda", os.path.join(__path__, "cuda.bc"), [],
+                                   {tuple(): ("globaltimer", core.dtype("int64")),
+                                    }, is_pure=False, _builder=_builder)
+
+
+@core.extern
+def smid(_builder=None):
+    return core.extern_elementwise("cuda", os.path.join(__path__, "cuda.bc"), [],
+                                   {tuple(): ("smid", core.dtype("int32")),
+                                    }, is_pure=True, _builder=_builder)