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 @@ -1,3 +1,21 @@ +if (MLIR_CUDA_CONVERSIONS_ENABLED) + set(NVPTX_LIBS + MC + NVPTXCodeGen + NVPTXDesc + NVPTXInfo + ) +endif() + +if (MLIR_ROCM_CONVERSIONS_ENABLED) + set(AMDGPU_LIBS + MC + AMDGPUCodeGen + AMDGPUDesc + AMDGPUInfo + ) +endif() + add_mlir_conversion_library(MLIRGPUtoGPURuntimeTransforms ConvertLaunchFuncToRuntimeCalls.cpp ConvertKernelFuncToBlob.cpp @@ -8,6 +26,8 @@ LINK_COMPONENTS Core + ${AMDGPU_LIBS} + ${NVPTX_LIBS} LINK_LIBS PUBLIC MLIRGPU diff --git a/mlir/test/lib/Transforms/CMakeLists.txt b/mlir/test/lib/Transforms/CMakeLists.txt --- a/mlir/test/lib/Transforms/CMakeLists.txt +++ b/mlir/test/lib/Transforms/CMakeLists.txt @@ -1,21 +1,3 @@ -if (MLIR_CUDA_CONVERSIONS_ENABLED) - set(NVPTX_LIBS - MC - NVPTXCodeGen - NVPTXDesc - NVPTXInfo - ) -endif() - -if (MLIR_ROCM_CONVERSIONS_ENABLED) - set(AMDGPU_LIBS - MC - AMDGPUCodeGen - AMDGPUDesc - AMDGPUInfo - ) -endif() - # Exclude tests from libMLIR.so add_mlir_library(MLIRTestTransforms TestAllReduceLowering.cpp @@ -50,10 +32,6 @@ MLIRStandardOpsIncGen MLIRTestVectorTransformPatternsIncGen - LINK_COMPONENTS - ${AMDGPU_LIBS} - ${NVPTX_LIBS} - LINK_LIBS PUBLIC MLIRAffineOps MLIRAnalysis