diff --git a/tinygrad/runtime/ops_python.py b/tinygrad/runtime/ops_python.py index 4bbf7af896..5615bb99c9 100644 --- a/tinygrad/runtime/ops_python.py +++ b/tinygrad/runtime/ops_python.py @@ -1,4 +1,4 @@ -# pylint: skip-file +# pylint: disable=cell-var-from-loop # a python uops emulator # works to test the tensor cores, and all the uops in general # this is the (living) definition of uops @@ -164,7 +164,7 @@ class PythonProgram: assert x[i][goff+j] == x[i][goff+j+16], "warp elements not duplicated properly across lanes" return x[i][goff+j] # B (16 elements on 32 threads): row major, lane 16-32 == lane 0-15 - def b_elem(x, i, j, goff): return a_elem(x, j, i, goff) + def b_elem(x, i, j, goff): return a_elem(x, j, i, goff) # pylint: disable=arguments-out-of-order def c_map(lane, elem): return (lane%16, lane//16+elem*2) # (i, j), C, D (8 elements on 32 threads): row major ul[i] = wmma_helper(32, 16, 16, 16, 8, a_elem, b_elem, c_map) elif arg[5] == "CUDA":