Skip to content

Commit 1c0bbb2

Browse files
committed
fix lint
1 parent 463574c commit 1c0bbb2

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

python/tvm/contrib/cutlass/gen_conv2d.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,6 @@
2222
from .conv2d_profiler import Conv2dProfilerEmitter
2323
from .gen_tensor_op import (
2424
ProfilerEngine,
25-
generate_sm75_tensor_op_1688,
26-
generate_sm80_tensor_op_16816,
2725
GENERATOR_FUNC_TABLE,
2826
)
2927
from .library import (
@@ -187,8 +185,10 @@ def profile(
187185
if profile_all:
188186
self.engine.compile_all(ops, use_multiprocessing)
189187

190-
args =("--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
191-
"--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d") % workload
188+
args = (
189+
"--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
190+
"--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d"
191+
) % workload
192192

193193
for op in ops:
194194
out = self.engine.evaluate(op, args.split(" "))

python/tvm/contrib/cutlass/gen_gemm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,6 @@
2222
from .gemm_profiler import GemmProfilerEmitter
2323
from .gen_tensor_op import (
2424
ProfilerEngine,
25-
generate_sm75_tensor_op_1688,
26-
generate_sm80_tensor_op_16816,
2725
GENERATOR_FUNC_TABLE,
2826
)
2927
from .library import (

0 commit comments

Comments
 (0)