deepcrayon/tinygrab
deepcrayon/tinygrab
1
0
Fork 0
Code Activity

Fix NOUNROLL (#1439)

Browse source
This commit is contained in:
Adrian Kretz 2023-08-05 01:50:19 +02:00 committed by GitHub
parent 579f4615a0
commit 043d5f2cb5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/codegen/cstyle.py
View file

@ -127,7 +127,7 @@ def uops_to_cstyle(uops:List[UOp], lang:CStyleLanguage) -> Tuple[str, List[int],
elif args[1] == "local" and lang.lid:
kk(add_gl_dimension(lang.size_prefix, args, i, var, local_size, lang.lid))
else:
if getenv("NOUNROLL"): kk("#pragma unroll(1)") # prevent loop unrolling
if getenv("NOUNROLL") and not isinstance(var, NumNode): kk("#pragma unroll(1)") # prevent loop unrolling
kk("{" if isinstance(var, NumNode) else lang.render_for(var.expr, var.min, var.max))
depth += 1
elif uop == UOps.BARRIER: