pylint fix

apache · Jan 18, 2021 · 01d6121 · 01d6121
1 parent 9843f5e
commit 01d6121
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/python/tvm/topi/cuda/scan.py b/python/tvm/topi/cuda/scan.py
@@ -14,6 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+# pylint: disable=invalid-name, too-many-locals, too-many-statements
 "Scan related operators"
 import tvm
 from tvm import te

diff --git a/python/tvm/topi/cuda/sparse.py b/python/tvm/topi/cuda/sparse.py
@@ -161,6 +161,7 @@ def sparse_dense_tir(data, w_data, w_indices, w_indptr):
     with either default_function_kernel0 for the transpose or
     default_function_kernel1 for the multiply.
     """
+
     def gen_ir(data, w_data, w_indices, w_indptr, out):
         # pylint: disable=invalid-name
         # TODO(tkonolige): use tensorcores for block multiply