From 4c7e316dedd252e7281d192a56ab3a43249daf71 Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 18 Jun 2024 20:15:34 -0400 Subject: [PATCH] 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. --- tinygrad/runtime/ops_python.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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":