Fix MLIR build when NVPTX backend is not configured in

The GPUToCUDA conversion needs to conditionally link it in.
2020-03-10 04:09:33 +00:00 · 2020-03-10 04:09:33 +00:00 · f80c6d8dec
parent 627bb31a28
commit f80c6d8dec
1 changed files with 6 additions and 3 deletions
--- a/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt
+++ b/mlir/lib/Conversion/GPUToCUDA/CMakeLists.txt
@ -8,14 +8,17 @@ set(SOURCES

 if (MLIR_CUDA_CONVERSIONS_ENABLED)
 list(APPEND SOURCES "ConvertKernelFuncToCubin.cpp")
+  set(NVPTX_LIBS
+    LLVMNVPTXCodeGen
+    LLVMNVPTXDesc
+    LLVMNVPTXInfo
+  )
 endif()

 add_mlir_conversion_library(MLIRGPUtoCUDATransforms ${SOURCES})
 target_link_libraries(MLIRGPUtoCUDATransforms
  PUBLIC
-  LLVMNVPTXCodeGen
-  LLVMNVPTXDesc
-  LLVMNVPTXInfo
+  ${NVPTX_LIBS}
  LLVMCore
  LLVMMC
  LLVMSupport