pylint

Author: masahi

python/tvm/relax/backend/contrib/cutlass.py

@@ -17,11 +17,11 @@
 
 """Pattern table for CUTLASS backend"""
 
-from typing import Mapping, Optional, Tuple
+from typing import Mapping, Optional
 
 import tvm
 from tvm.contrib.cutlass.build import is_shape_valid_for_cutlass_matmul
-from tvm.relax import Call, Expr, ShapeExpr, transform
+from tvm.relax import Call, Expr, transform
 from tvm.relax.dpl import CallPattern, DFPattern
 
 from ..pattern_registry import get_patterns_with_prefix, register_patterns

python/tvm/relax/testing/matmul.py

@@ -14,6 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+"""Utilities to construct matmul workloads."""
 import tvm
 from tvm.script import relax as R
 from tvm.script.ir_builder import IRBuilder
@@ -23,6 +24,7 @@
 def get_relax_matmul_module(
     x_shape, y_shape, dtype, transposed_y=False, with_bias=False, activation=None
 ):
+    """Create a module with matmul op followed by activation."""
     if transposed_y:
         n = y_shape[-2]
     else: