diff --git a/mlir/lib/CAPI/Dialect/CMakeLists.txt b/mlir/lib/CAPI/Dialect/CMakeLists.txt --- a/mlir/lib/CAPI/Dialect/CMakeLists.txt +++ b/mlir/lib/CAPI/Dialect/CMakeLists.txt @@ -23,7 +23,7 @@ LINK_LIBS PUBLIC MLIRCAPIIR - MLIRGPU + MLIRGPUTransforms MLIRPass ) diff --git a/mlir/lib/Conversion/GPUCommon/CMakeLists.txt b/mlir/lib/Conversion/GPUCommon/CMakeLists.txt --- a/mlir/lib/Conversion/GPUCommon/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUCommon/CMakeLists.txt @@ -30,7 +30,7 @@ LINK_LIBS PUBLIC MLIRAsyncToLLVM - MLIRGPU + MLIRGPUTransforms MLIRIR MLIRLLVMIR MLIRPass diff --git a/mlir/lib/Conversion/GPUToNVVM/CMakeLists.txt b/mlir/lib/Conversion/GPUToNVVM/CMakeLists.txt --- a/mlir/lib/Conversion/GPUToNVVM/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUToNVVM/CMakeLists.txt @@ -11,7 +11,7 @@ MLIRGPUToNVVMIncGen LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRGPUToGPURuntimeTransforms MLIRLLVMIR MLIRMemRef diff --git a/mlir/lib/Conversion/GPUToROCDL/CMakeLists.txt b/mlir/lib/Conversion/GPUToROCDL/CMakeLists.txt --- a/mlir/lib/Conversion/GPUToROCDL/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUToROCDL/CMakeLists.txt @@ -10,7 +10,7 @@ MLIRGPUToROCDLIncGen LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRGPUToGPURuntimeTransforms MLIRLLVMIR MLIRROCDLIR diff --git a/mlir/lib/Conversion/GPUToSPIRV/CMakeLists.txt b/mlir/lib/Conversion/GPUToSPIRV/CMakeLists.txt --- a/mlir/lib/Conversion/GPUToSPIRV/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUToSPIRV/CMakeLists.txt @@ -6,7 +6,7 @@ MLIRConversionPassIncGen LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRIR MLIRPass MLIRSCFToSPIRV diff --git a/mlir/lib/Conversion/GPUToVulkan/CMakeLists.txt b/mlir/lib/Conversion/GPUToVulkan/CMakeLists.txt --- a/mlir/lib/Conversion/GPUToVulkan/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUToVulkan/CMakeLists.txt @@ -6,7 +6,7 @@ MLIRConversionPassIncGen LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRIR MLIRLLVMIR MLIRPass diff --git a/mlir/lib/Conversion/SCFToGPU/CMakeLists.txt b/mlir/lib/Conversion/SCFToGPU/CMakeLists.txt --- a/mlir/lib/Conversion/SCFToGPU/CMakeLists.txt +++ b/mlir/lib/Conversion/SCFToGPU/CMakeLists.txt @@ -12,7 +12,7 @@ MLIRAffine MLIRAffineToStandard MLIRComplex - MLIRGPU + MLIRGPUTransforms MLIRIR MLIRLinalg MLIRMemRef diff --git a/mlir/lib/Conversion/SPIRVToLLVM/CMakeLists.txt b/mlir/lib/Conversion/SPIRVToLLVM/CMakeLists.txt --- a/mlir/lib/Conversion/SPIRVToLLVM/CMakeLists.txt +++ b/mlir/lib/Conversion/SPIRVToLLVM/CMakeLists.txt @@ -11,7 +11,7 @@ intrinsics_gen LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRSPIRV MLIRSPIRVUtils MLIRLLVMIR diff --git a/mlir/lib/Conversion/VectorToGPU/CMakeLists.txt b/mlir/lib/Conversion/VectorToGPU/CMakeLists.txt --- a/mlir/lib/Conversion/VectorToGPU/CMakeLists.txt +++ b/mlir/lib/Conversion/VectorToGPU/CMakeLists.txt @@ -8,7 +8,7 @@ Core LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRLLVMIR MLIRMemRef MLIRTransforms diff --git a/mlir/lib/Dialect/GPU/CMakeLists.txt b/mlir/lib/Dialect/GPU/CMakeLists.txt --- a/mlir/lib/Dialect/GPU/CMakeLists.txt +++ b/mlir/lib/Dialect/GPU/CMakeLists.txt @@ -16,8 +16,25 @@ ) endif() -add_mlir_dialect_library(MLIRGPU +add_mlir_dialect_library(MLIRGPUOps IR/GPUDialect.cpp + + ADDITIONAL_HEADER_DIRS + ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/GPU + + DEPENDS + MLIRGPUOpsIncGen + MLIRGPUOpInterfacesIncGen + + LINK_LIBS PUBLIC + MLIRDLTI + MLIRIR + MLIRMemRef + MLIRSideEffectInterfaces + MLIRSupport + ) + +add_mlir_dialect_library(MLIRGPUTransforms Transforms/AllReduceLowering.cpp Transforms/AsyncRegionRewriter.cpp Transforms/KernelOutlining.cpp @@ -37,8 +54,6 @@ ${AMDGPU_LIBS} DEPENDS - MLIRGPUOpsIncGen - MLIRGPUOpInterfacesIncGen MLIRGPUPassIncGen MLIRParallelLoopMapperAttrGen MLIRParallelLoopMapperEnumsGen @@ -46,14 +61,13 @@ LINK_LIBS PUBLIC MLIRAsync MLIRDataLayoutInterfaces - MLIRDLTI + MLIRGPUOps MLIRIR - MLIRMemRef MLIRLLVMIR MLIRLLVMToLLVMIRTranslation - MLIRSCF + MLIRMemRef MLIRPass - MLIRSideEffectInterfaces + MLIRSCF MLIRStandard MLIRSupport MLIRTransformUtils @@ -77,20 +91,20 @@ endif() # Enable gpu-to-cubin pass. - target_compile_definitions(obj.MLIRGPU + target_compile_definitions(obj.MLIRGPUTransforms PRIVATE MLIR_GPU_TO_CUBIN_PASS_ENABLE=1 ) # Add CUDA headers includes and the libcuda.so library. - target_include_directories(obj.MLIRGPU + target_include_directories(obj.MLIRGPUTransforms PRIVATE ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES} ) find_library(CUDA_DRIVER_LIBRARY cuda) - target_link_libraries(MLIRGPU + target_link_libraries(MLIRGPUTransforms PRIVATE MLIRNVVMToLLVMIRTranslation ${CUDA_DRIVER_LIBRARY} @@ -126,21 +140,21 @@ message(STATUS "ROCm HIP version: ${HIP_VERSION}") endif() - target_compile_definitions(obj.MLIRGPU + target_compile_definitions(obj.MLIRGPUOps PRIVATE __HIP_PLATFORM_HCC__ __ROCM_PATH__="${ROCM_PATH}" MLIR_GPU_TO_HSACO_PASS_ENABLE=1 ) - target_include_directories(obj.MLIRGPU + target_include_directories(obj.MLIRGPUOps PRIVATE ${MLIR_SOURCE_DIR}/../lld/include ${HIP_PATH}/include ${ROCM_PATH}/include ) - target_link_libraries(MLIRGPU + target_link_libraries(MLIRGPUOps PRIVATE lldELF MLIRROCDLToLLVMIRTranslation diff --git a/mlir/test/lib/Dialect/GPU/CMakeLists.txt b/mlir/test/lib/Dialect/GPU/CMakeLists.txt --- a/mlir/test/lib/Dialect/GPU/CMakeLists.txt +++ b/mlir/test/lib/Dialect/GPU/CMakeLists.txt @@ -10,7 +10,7 @@ LINK_LIBS PUBLIC MLIRAffine - MLIRGPU + MLIRGPUOps MLIRGPUToGPURuntimeTransforms MLIRLLVMToLLVMIRTranslation MLIRNVVMIR diff --git a/mlir/test/lib/Dialect/Linalg/CMakeLists.txt b/mlir/test/lib/Dialect/Linalg/CMakeLists.txt --- a/mlir/test/lib/Dialect/Linalg/CMakeLists.txt +++ b/mlir/test/lib/Dialect/Linalg/CMakeLists.txt @@ -12,7 +12,7 @@ LINK_LIBS PUBLIC MLIRAffine - MLIRGPU + MLIRGPUTransforms MLIRLinalg MLIRLinalgTransforms MLIRLLVMToLLVMIRTranslation diff --git a/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt b/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt --- a/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt +++ b/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt @@ -12,7 +12,7 @@ ${MLIR_MAIN_INCLUDE_DIR}/mlir/IR LINK_LIBS PUBLIC - MLIRGPU + MLIRGPUOps MLIRIR MLIRPass MLIRSPIRV diff --git a/mlir/tools/mlir-spirv-cpu-runner/CMakeLists.txt b/mlir/tools/mlir-spirv-cpu-runner/CMakeLists.txt --- a/mlir/tools/mlir-spirv-cpu-runner/CMakeLists.txt +++ b/mlir/tools/mlir-spirv-cpu-runner/CMakeLists.txt @@ -17,7 +17,7 @@ ${conversion_libs} MLIRAnalysis MLIRExecutionEngine - MLIRGPU + MLIRGPUOps MLIRIR MLIRJitRunner MLIRLLVMIR diff --git a/mlir/tools/mlir-vulkan-runner/CMakeLists.txt b/mlir/tools/mlir-vulkan-runner/CMakeLists.txt --- a/mlir/tools/mlir-vulkan-runner/CMakeLists.txt +++ b/mlir/tools/mlir-vulkan-runner/CMakeLists.txt @@ -57,7 +57,7 @@ ${conversion_libs} MLIRAnalysis MLIRExecutionEngine - MLIRGPU + MLIRGPUOps MLIRIR MLIRJitRunner MLIRLLVMIR