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,4 +1,10 @@ # Exclude tests from libMLIR.so +if (MLIR_CUDA_CONVERSIONS_ENABLED) + set(GPU_TO_CUDA_LIB "MLIRGPUtoCUDATransforms") +else() + set(GPU_TO_CUDA_LIB "") +endif() + add_mlir_library(MLIRTestTransforms TestAllReduceLowering.cpp TestBufferPlacement.cpp @@ -36,7 +42,7 @@ MLIRAnalysis MLIREDSC MLIRGPU - MLIRGPUtoCUDATransforms + ${GPU_TO_CUDA_LIB} MLIRLinalgOps MLIRLinalgTransforms MLIRSCF