2020-02-27 03:50:03 +08:00
|
|
|
add_mlir_dialect_library(MLIRGPU
|
2019-04-29 18:00:25 +08:00
|
|
|
IR/GPUDialect.cpp
|
[mlir] Add in-dialect lowering of gpu.all_reduce.
Reviewers: ftynse, nicolasvasilache, herhut
Reviewed By: ftynse, herhut
Subscribers: liufengdb, aartbik, herhut, merge_guards_bot, mgorny, mehdi_amini, rriddle, jpienaar, burmako, shauheen, antiagainst, nicolasvasilache, arpith-jacob, mgester, lucyrfox, llvm-commits
Tags: #llvm
Differential Revision: https://reviews.llvm.org/D72129
2020-01-20 20:30:24 +08:00
|
|
|
Transforms/AllReduceLowering.cpp
|
2019-05-23 17:16:18 +08:00
|
|
|
Transforms/KernelOutlining.cpp
|
2020-01-08 03:00:54 +08:00
|
|
|
Transforms/MemoryPromotion.cpp
|
2020-02-21 23:18:22 +08:00
|
|
|
Transforms/ParallelLoopMapper.cpp
|
2019-04-29 18:00:25 +08:00
|
|
|
|
|
|
|
ADDITIONAL_HEADER_DIRS
|
2019-07-25 15:40:48 +08:00
|
|
|
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/GPU
|
2020-02-21 02:48:51 +08:00
|
|
|
|
|
|
|
DEPENDS
|
2020-03-01 03:51:04 +08:00
|
|
|
MLIRGPUOpsIncGen
|
2020-03-25 06:55:36 +08:00
|
|
|
MLIRParallelLoopMapperAttrGen
|
|
|
|
MLIRParallelLoopMapperEnumsGen
|
2020-02-20 14:56:38 +08:00
|
|
|
)
|
2020-03-01 03:51:04 +08:00
|
|
|
target_link_libraries(MLIRGPU
|
2020-03-06 03:49:42 +08:00
|
|
|
PUBLIC
|
2020-03-01 03:50:45 +08:00
|
|
|
MLIREDSC
|
|
|
|
MLIRIR
|
|
|
|
MLIRLLVMIR
|
|
|
|
MLIRLoopOps
|
2020-03-01 03:51:04 +08:00
|
|
|
MLIRPass
|
2020-03-11 03:25:08 +08:00
|
|
|
MLIRSideEffects
|
2020-03-01 03:51:04 +08:00
|
|
|
MLIRStandardOps
|
2020-03-01 03:50:45 +08:00
|
|
|
MLIRSupport
|
|
|
|
MLIRTransformUtils
|
2020-02-21 02:48:51 +08:00
|
|
|
LLVMSupport
|
|
|
|
)
|