File tree Expand file tree Collapse file tree 3 files changed +1
-16
lines changed
python/tvm/contrib/cutlass Expand file tree Collapse file tree 3 files changed +1
-16
lines changed Original file line number Diff line number Diff line change 1818"""Conv2d kernel generator and profiler for CUTLASS."""
1919from .conv2d_operation import Conv2dOperation, EmitConv2dInstance
2020from .conv2d_profiler import Conv2dProfilerEmitter
21- from .gemm_profiler import GemmProfilerEmitter
2221from gen_tensor_op import (
2322 ProfilerEngine,
2423 generate_sm75_tensor_op_1688,
3029 TensorDescription,
3130 DataTypeTag,
3231 LayoutType,
33- MathInstruction,
34- DataType,
35- OpcodeClass,
36- MathOperation,
37- TileDescription,
3832 ConvKind,
39- IteratorAlgorithm,
4033 StrideSupport,
34+ IteratorAlgorithm
4135)
4236
4337
Original file line number Diff line number Diff line change 3434 TensorDescription,
3535 DataTypeTag,
3636 LayoutType,
37- MathInstruction,
38- DataType,
39- OpcodeClass,
40- MathOperation,
41- TileDescription,
4237)
4338
4439
Original file line number Diff line number Diff line change 2323import subprocess
2424import multiprocessing
2525from .library import (
26- EpilogueFunctor,
27- SwizzlingFunctor,
2826 TensorDescription,
29- DataTypeTag,
30- LayoutType,
3127 MathInstruction,
3228 DataType,
3329 OpcodeClass,
You can’t perform that action at this time.
0 commit comments