1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
add_mlir_dialect_library(MLIRSparseTensorTransforms
# Rewriting.
BufferizableOpInterfaceImpl.cpp
SparseAssembler.cpp
SparseBufferRewriting.cpp
SparseGPUCodegen.cpp
SparseIterationToScf.cpp
SparseReinterpretMap.cpp
SparseStorageSpecifierToLLVM.cpp
SparseSpaceCollapse.cpp
SparseTensorCodegen.cpp
SparseTensorConversion.cpp
SparseTensorPasses.cpp
SparseTensorRewriting.cpp
SparseVectorization.cpp
Sparsification.cpp
SparsificationAndBufferizationPass.cpp
StageSparseOperations.cpp
# Utilities.
Utils/CodegenEnv.cpp
Utils/CodegenUtils.cpp
Utils/IterationGraphSorter.cpp
Utils/LoopEmitter.cpp
Utils/SparseTensorDescriptor.cpp
Utils/SparseTensorIterator.cpp
ADDITIONAL_HEADER_DIRS
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/SparseTensor
DEPENDS
MLIRSparseTensorPassIncGen
LINK_LIBS PUBLIC
MLIRArithDialect
MLIRBufferizationDialect
MLIRBufferizationTransforms
MLIRComplexDialect
MLIRFuncDialect
MLIRIR
MLIRLLVMDialect
MLIRLinalgDialect
MLIRLinalgTransforms
MLIRMemRefDialect
MLIRPass
MLIRSCFDialect
MLIRSCFTransforms
MLIRSCFUtils
MLIRSparseTensorDialect
MLIRSparseTensorEnums
MLIRSparseTensorUtils
MLIRTensorDialect
MLIRTransforms
MLIRVectorDialect
)
|
