diff --git a/python/tvm/contrib/cutlass/gen_conv2d.py b/python/tvm/contrib/cutlass/gen_conv2d.py index 24d6c2b1e639b..a5beb357b309c 100644 --- 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 index 65d460084ca69..7287d24718177 100644 --- 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 (