diff --git a/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt b/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt --- a/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt +++ b/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt @@ -1,16 +1,11 @@ -if(MLIR_CUDA_CONVERSIONS_ENABLED) - add_llvm_library(MLIRGPUtoCUDATransforms - ConvertKernelFuncToCubin.cpp - ConvertLaunchFuncToCudaCalls.cpp - ) - target_link_libraries(MLIRGPUtoCUDATransforms - LLVMNVPTXCodeGen - LLVMNVPTXDesc - LLVMNVPTXInfo - MLIRGPU - MLIRLLVMIR - MLIRNVVMIR - MLIRPass - MLIRTargetNVVMIR - ) -endif() +add_llvm_library(MLIRGPUtoCUDATransforms + ConvertKernelFuncToCubin.cpp + ConvertLaunchFuncToCudaCalls.cpp +) +target_link_libraries(MLIRGPUtoCUDATransforms + MLIRGPU + MLIRLLVMIR + MLIRNVVMIR + MLIRPass + MLIRTargetNVVMIR +) diff --git a/mlir/lib/Conversion/GPUToCUDA/ConvertKernelFuncToCubin.cpp b/mlir/lib/Conversion/GPUToCUDA/ConvertKernelFuncToCubin.cpp --- a/mlir/lib/Conversion/GPUToCUDA/ConvertKernelFuncToCubin.cpp +++ b/mlir/lib/Conversion/GPUToCUDA/ConvertKernelFuncToCubin.cpp @@ -57,10 +57,9 @@ gpu::GPUModuleOp module = getOperation(); // Make sure the NVPTX target is initialized. - LLVMInitializeNVPTXTarget(); - LLVMInitializeNVPTXTargetInfo(); - LLVMInitializeNVPTXTargetMC(); - LLVMInitializeNVPTXAsmPrinter(); + llvm::InitializeAllTargets(); + llvm::InitializeAllTargetMCs(); + llvm::InitializeAllAsmPrinters(); auto llvmModule = translateModuleToNVVMIR(module); if (!llvmModule) diff --git a/mlir/tools/mlir-opt/CMakeLists.txt b/mlir/tools/mlir-opt/CMakeLists.txt --- a/mlir/tools/mlir-opt/CMakeLists.txt +++ b/mlir/tools/mlir-opt/CMakeLists.txt @@ -31,6 +31,7 @@ MLIREDSC MLIRFxpMathOps MLIRGPU + MLIRGPUtoCUDATransforms MLIRGPUtoNVVMTransforms MLIRGPUtoROCDLTransforms MLIRGPUtoSPIRVTransforms @@ -75,15 +76,10 @@ LLVMAsmParser ) -if(MLIR_CUDA_CONVERSIONS_ENABLED) - list(APPEND LIBS - LLVMNVPTXCodeGen - LLVMNVPTXDesc - LLVMNVPTXInfo - MLIRGPUtoCUDATransforms - MLIRTargetNVVMIR - ) -endif() +set(LLVM_LINK_COMPONENTS + AllTargetsCodeGens +) + add_llvm_tool(mlir-opt mlir-opt.cpp )