github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-18 18:35:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

indexing merge almost works

Browse Source
This commit is contained in:
George Hotz
2023-03-20 16:17:07 -07:00
parent 06abbbfe7c
commit 120d7072bd
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/codegen/cstyle.py
View File

@@ -186,12 +186,12 @@ class CStyleCodegen(Linearizer):
def codegen(self):
self.process()
self.hand_coded_optimizations()
# sometimes, there's more dimensions than len(self.lang.gid).
# compact all the dimensions into the first
# NOTE: this might make multiview shapetrackers
# NOTE: you ABSOLUTELY must do this before upcasting. the strides on the upcast are wrong if you don't
# TODO: this exposes bugs in the optimizers assuming the strides are on a single view
# TODO: this exposes bugs in the optimizers assuming the strides are on a single vie
"""
if len(self.lang.gid) and self.first_reduce > len(self.lang.gid):
num_to_merge = (self.first_reduce - len(self.lang.gid))+1
@@ -199,7 +199,6 @@ class CStyleCodegen(Linearizer):
if DEBUG >= 4: print("reshaped to", self.full_shape, "due to too many global dimensions")
"""
self.hand_coded_optimizations()
self.linearize()
prg, global_size, local_size = uops_to_cstyle(self.uops, self.bufs, self.lang)