forked from tinygrad/tinygrad
update pylint for ops_python (#5046)
the two errors (cell-var-from-loop and arguments-out-of-order) does not apply as we use it as intended.
This commit is contained in:
@@ -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":
|
||||
|
||||
Reference in New Issue
Block a user