Skip to content

Commit 68b7d3f

Browse files
[mlir][gpu][transforms] Only depend on ExecutionEngine if MLIR_ENABLE_CUDA_RUNNER is true
This fixes a compilation bug where we would try to depend on ExecutionEngine but it wasn't actually built.
1 parent 90ecfa2 commit 68b7d3f

File tree

1 file changed

+16
-11
lines changed

1 file changed

+16
-11
lines changed

mlir/test/lib/Dialect/GPU/CMakeLists.txt

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ set(LIBS
88
MLIRAnalysis
99
MLIRArithDialect
1010
MLIRBuiltinToLLVMIRTranslation
11-
MLIRExecutionEngine
1211
MLIRFuncDialect
1312
MLIRGPUDialect
1413
MLIRGPUToGPURuntimeTransforms
@@ -35,6 +34,22 @@ set(LIBS
3534
MLIRVectorToLLVM
3635
)
3736

37+
# This is how it is defined in mlir/lib/Dialect/GPU/CMakeLists.txt
38+
# We probably want something better project-wide
39+
if(MLIR_ENABLE_CUDA_RUNNER)
40+
# Enable gpu-to-cubin pass.
41+
target_compile_definitions(MLIRGPUTestPasses
42+
PRIVATE
43+
MLIR_GPU_TO_CUBIN_PASS_ENABLE=1
44+
)
45+
46+
set(LIBS
47+
${LIBS}
48+
49+
MLIRExecutionEngine
50+
)
51+
endif()
52+
3853
add_mlir_library(MLIRGPUTestPasses
3954
TestConvertGPUKernelToCubin.cpp
4055
TestConvertGPUKernelToHsaco.cpp
@@ -47,13 +62,3 @@ add_mlir_library(MLIRGPUTestPasses
4762
LINK_LIBS PUBLIC
4863
${LIBS}
4964
)
50-
51-
# This is how it is defined in mlir/lib/Dialect/GPU/CMakeLists.txt
52-
# We probably want something better project-wide
53-
if(MLIR_ENABLE_CUDA_RUNNER)
54-
# Enable gpu-to-cubin pass.
55-
target_compile_definitions(MLIRGPUTestPasses
56-
PRIVATE
57-
MLIR_GPU_TO_CUBIN_PASS_ENABLE=1
58-
)
59-
endif()

0 commit comments

Comments
 (0)