[3] lint error resolved

apache · Nov 9, 2020 · 06f9911 · 06f9911
1 parent bb733b4
commit 06f9911
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py
@@ -1993,6 +1993,7 @@ def batch_matmul(x, y):
     return _make.batch_matmul(x, y)
 
 
+# pylint: disable=no-else-return,inconsistent-return-statements
 def sparse_dense(data, weight, sparse_data=False):
     r"""
     Computes the matrix multiplication of `data` and `weight`, where `data` is

diff --git a/python/tvm/topi/nn/sparse.py b/python/tvm/topi/nn/sparse.py
@@ -93,6 +93,7 @@ def sparse_dense_v1(data_data, data_indices, data_indptr, weight):
     return func(data_data, data_indices, data_indptr, weight)
 
 
+# pylint: disable=no-else-return,inconsistent-return-statements
 def sparse_dense(input_1, input_2, input_3, input_4, sparse_data=False):
     """
     Computes sparse-dense matrix multiplication of `data` and
@@ -181,8 +182,8 @@ def f(i, row):
 
 
 def _sparse_dense_bsrmm_v1(data_data, data_indices, data_indptr, weight):
-    (k, m) = get_const_tuple(weight.shape)
-    (l, bs_r, bs_c) = get_const_tuple(data_data.shape)
+    (k, _) = get_const_tuple(weight.shape)
+    (_, bs_r, bs_c) = get_const_tuple(data_data.shape)
     (num_blocks_plus_1,) = get_const_tuple(data_indptr.shape)
     num_blocks = num_blocks_plus_1 - 1