github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

llvm was using wrong shapetracker

Browse Source
This commit is contained in:
George Hotz
2023-03-12 11:49:03 -07:00
parent fdde87afda
commit 46b49d50bd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/codegen/llvm.py
View File

@@ -137,7 +137,7 @@ class LLVMCodegen(ASTKernel):
buf_index = self.bufs.index(x)
for i, idx in enumerate(get_idxs(builder, idx_level[buf_index][level], buf_index)):
# first view is already implictly handled
idx, valid = x.st._expr_idx(Variable(idx, 0, prod(x.st.shape)))
idx, valid = self.sts[buf_index]._expr_idx(Variable(idx, 0, prod(self.sts[buf_index].shape)))
idx = idx.render(render_llvm, builder)
if valid.min == 0:
valid = valid.render(render_llvm, builder)