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