0b1764a3d7
This revision connects the generated sparse code with an actual sparse storage scheme, which can be initialized from a test file. Lacking a first-class citizen SparseTensor type (with buffer), the storage is hidden behind an opaque pointer with some "glue" to bring the pointer back to tensor land. Rather than generating sparse setup code for each different annotated tensor (viz. the "pack" methods in TACO), a single "one-size-fits-all" implementation has been added to the runtime support library. Many details and abstractions need to be refined in the future, but this revision allows full end-to-end integration testing and performance benchmarking (with on one end, an annotated Lingalg op and, on the other end, a JIT/AOT executable). Reviewed By: nicolasvasilache, bixia Differential Revision: https://reviews.llvm.org/D95847
47 lines
789 B
CMake
47 lines
789 B
CMake
add_mlir_dialect_library(MLIRLinalgTransforms
|
|
Bufferize.cpp
|
|
CodegenStrategy.cpp
|
|
DropUnitDims.cpp
|
|
ElementwiseToLinalg.cpp
|
|
Fusion.cpp
|
|
FusionOnTensors.cpp
|
|
Generalization.cpp
|
|
Hoisting.cpp
|
|
Interchange.cpp
|
|
Loops.cpp
|
|
Promotion.cpp
|
|
SparseLowering.cpp
|
|
Sparsification.cpp
|
|
Tiling.cpp
|
|
Transforms.cpp
|
|
Vectorization.cpp
|
|
|
|
ADDITIONAL_HEADER_DIRS
|
|
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Linalg
|
|
|
|
DEPENDS
|
|
MLIRLinalgPassIncGen
|
|
|
|
LINK_LIBS PUBLIC
|
|
MLIRAffine
|
|
MLIRAffineUtils
|
|
MLIRAnalysis
|
|
MLIREDSC
|
|
MLIRIR
|
|
MLIRLinalgAnalysis
|
|
MLIRLinalgEDSC
|
|
MLIRLinalg
|
|
MLIRLinalgUtils
|
|
MLIRSCF
|
|
MLIRSCFTransforms
|
|
MLIRPass
|
|
MLIRStandard
|
|
MLIRStandardOpsTransforms
|
|
MLIRStandardToLLVM
|
|
MLIRTensor
|
|
MLIRTransforms
|
|
MLIRTransformUtils
|
|
MLIRVector
|
|
MLIRVectorToSCF
|
|
)
|