remove unused import

mehrdadh · Nov 13, 2021 · 1a6b27c · 1a6b27c
1 parent f7c3b5a
commit 1a6b27c
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 16 deletions.
diff --git 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
@@ -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
@@ -23,11 +23,7 @@
 import subprocess
 import multiprocessing
 from .library import (
-    EpilogueFunctor,
-    SwizzlingFunctor,
     TensorDescription,
-    DataTypeTag,
-    LayoutType,
     MathInstruction,
     DataType,
     OpcodeClass,