fix lint

yangulei · Dec 12, 2021 · 1c0bbb2 · 1c0bbb2
1 parent 463574c
commit 1c0bbb2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/python/tvm/contrib/cutlass/gen_conv2d.py b/python/tvm/contrib/cutlass/gen_conv2d.py
@@ -22,8 +22,6 @@
 from .conv2d_profiler import Conv2dProfilerEmitter
 from .gen_tensor_op import (
     ProfilerEngine,
-    generate_sm75_tensor_op_1688,
-    generate_sm80_tensor_op_16816,
     GENERATOR_FUNC_TABLE,
 )
 from .library import (
@@ -187,8 +185,10 @@ def profile(
         if profile_all:
             self.engine.compile_all(ops, use_multiprocessing)
 
-        args =("--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
-               "--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d") % workload
+        args = (
+            "--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
+            "--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d"
+        ) % workload
 
         for op in ops:
             out = self.engine.evaluate(op, args.split(" "))

diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py
@@ -22,8 +22,6 @@
 from .gemm_profiler import GemmProfilerEmitter
 from .gen_tensor_op import (
     ProfilerEngine,
-    generate_sm75_tensor_op_1688,
-    generate_sm80_tensor_op_16816,
     GENERATOR_FUNC_TABLE,
 )
 from .library import (