diff --git a/python/tvm/contrib/cutlass/gen_conv2d.py b/python/tvm/contrib/cutlass/gen_conv2d.py index ee9017439d8f9..6fd77e4c2b6bb 100644 --- a/python/tvm/contrib/cutlass/gen_conv2d.py +++ b/python/tvm/contrib/cutlass/gen_conv2d.py @@ -18,7 +18,6 @@ """Conv2d kernel generator and profiler for CUTLASS.""" from .conv2d_operation import Conv2dOperation, EmitConv2dInstance from .conv2d_profiler import Conv2dProfilerEmitter -from .gemm_profiler import GemmProfilerEmitter from gen_tensor_op import ( ProfilerEngine, generate_sm75_tensor_op_1688, @@ -30,14 +29,9 @@ TensorDescription, DataTypeTag, LayoutType, - MathInstruction, - DataType, - OpcodeClass, - MathOperation, - TileDescription, ConvKind, - IteratorAlgorithm, StrideSupport, + IteratorAlgorithm ) diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py index 9dee41c82794e..5a023a3eb5186 100644 --- a/python/tvm/contrib/cutlass/gen_gemm.py +++ b/python/tvm/contrib/cutlass/gen_gemm.py @@ -34,11 +34,6 @@ TensorDescription, DataTypeTag, LayoutType, - MathInstruction, - DataType, - OpcodeClass, - MathOperation, - TileDescription, ) diff --git a/python/tvm/contrib/cutlass/gen_tensor_op.py b/python/tvm/contrib/cutlass/gen_tensor_op.py index 1fee256a731dd..68bbc94d67770 100644 --- a/python/tvm/contrib/cutlass/gen_tensor_op.py +++ b/python/tvm/contrib/cutlass/gen_tensor_op.py @@ -23,11 +23,7 @@ import subprocess import multiprocessing from .library import ( - EpilogueFunctor, - SwizzlingFunctor, TensorDescription, - DataTypeTag, - LayoutType, MathInstruction, DataType, OpcodeClass,