From 46b49d50bd352571189db1caf72dfb60e370ac3d Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 12 Mar 2023 11:49:03 -0700 Subject: [PATCH] llvm was using wrong shapetracker --- tinygrad/codegen/llvm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/codegen/llvm.py b/tinygrad/codegen/llvm.py index 2bb62409e4..ddf3afabd7 100644 --- a/tinygrad/codegen/llvm.py +++ b/tinygrad/codegen/llvm.py @@ -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)