add_llvm_library(MLIRGPU
  IR/GPUDialect.cpp
  IR/DialectRegistration.cpp
  Transforms/KernelOutlining.cpp
  Transforms/MemoryPromotion.cpp

  ADDITIONAL_HEADER_DIRS
  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/GPU
)
add_dependencies(MLIRGPU
  MLIRGPUOpsIncGen
  MLIREDSC
  MLIRIR
  MLIRLLVMIR
  MLIRLoopOps
  MLIRSupport
  MLIRTransformUtils
  LLVMSupport)
target_link_libraries(MLIRGPU
  MLIREDSC
  MLIRIR
  MLIRLLVMIR
  MLIRLoopOps
  MLIRSupport
  MLIRTransformUtils
  LLVMSupport)