tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-09-06 07:06:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

compile cooperative AMD kernels at O2

Browse Source
This commit is contained in:
geohot
2026-07-12 12:03:08 +00:00
parent a030286672
commit 2908c33a13
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/support/compiler_amd.py
+1 -1
View File
@@ -103,7 +103,7 @@ class HIPCompiler(Compiler):
except RuntimeError as e: raise CompileError(e) from e
def compile_cached_batch(self, srcs:list[tuple[str, str]]) -> list[tuple[str, str, bytes]]:
batch_size = getenv("AMD_COMPILE_BATCH_SIZE", 256)
batch_opt = getenv("AMD_COMPILE_OPT", 3)
batch_opt = getenv("AMD_COMPILE_OPT", 2)
generic_opt = getenv("AMD_GENERIC_COMPILE_OPT", 1)
frontend_opt = getenv("AMD_FRONTEND_OPT", 1)
if batch_size <= 1 or any(src.split('\n', 1)[0].strip() == '.text' for _,src in srcs): return super().compile_cached_batch(srcs)