nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

gep tuple [run_process_replay] (#6495)

Browse Source 
* gep tuple [run_process_replay]

* no inf loop, that goes in expander

* fix ops python

* unbreak gep 0

* fix tests

* fix tests

* VECTORIZE/GEP

* oops, broken
This commit is contained in:
George Hotz
2024-09-12 16:37:31 +08:00
committed by GitHub
parent 6dfa63cb21
commit a532d59bbd
7 changed files with 41 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_python.py
+2 -1
View File
@@ -127,7 +127,8 @@ class PythonProgram:
for j in range(len(inp[0])): inp[0][j] = inp[1][j]
ul[i] = inp[0]
elif uop is UOps.GEP:
ul[i] = inp[0][arg]
assert len(arg) == 1
ul[i] = inp[0][arg[0]]
elif uop is UOps.WMMA:
# here are the models for the WMMA instruction on the different hardware
def wmma_helper(WARP_THREADS, K, NUM_A, NUM_B, NUM_C, a_elem, b_elem, c_map):