add_mlir_dialect_library(MLIRGPU | |
IR/GPUDialect.cpp | |
Transforms/AllReduceLowering.cpp | |
Transforms/KernelOutlining.cpp | |
Transforms/MemoryPromotion.cpp | |
Transforms/ParallelLoopMapper.cpp | |
ADDITIONAL_HEADER_DIRS | |
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/GPU | |
DEPENDS | |
MLIRGPUOpsIncGen | |
MLIRGPUPassIncGen | |
MLIRParallelLoopMapperAttrGen | |
MLIRParallelLoopMapperEnumsGen | |
LINK_LIBS PUBLIC | |
MLIREDSC | |
MLIRIR | |
MLIRLLVMIR | |
MLIRSCF | |
MLIRPass | |
MLIRSideEffectInterfaces | |
MLIRStandardOps | |
MLIRSupport | |
MLIRTransformUtils | |
) |