File tree Expand file tree Collapse file tree 1 file changed +16
-11
lines changed
mlir/test/lib/Dialect/GPU Expand file tree Collapse file tree 1 file changed +16
-11
lines changed Original file line number Diff line number Diff line change 8
8
MLIRAnalysis
9
9
MLIRArithDialect
10
10
MLIRBuiltinToLLVMIRTranslation
11
- MLIRExecutionEngine
12
11
MLIRFuncDialect
13
12
MLIRGPUDialect
14
13
MLIRGPUToGPURuntimeTransforms
@@ -35,6 +34,22 @@ set(LIBS
35
34
MLIRVectorToLLVM
36
35
)
37
36
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
+
38
53
add_mlir_library (MLIRGPUTestPasses
39
54
TestConvertGPUKernelToCubin.cpp
40
55
TestConvertGPUKernelToHsaco.cpp
@@ -47,13 +62,3 @@ add_mlir_library(MLIRGPUTestPasses
47
62
LINK_LIBS PUBLIC
48
63
${LIBS}
49
64
)
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 ()
You can’t perform that action at this time.
0 commit comments