Skip to content

Commit 9d58f89

Browse files
Add symlink for FlangEnzyme, when ClangEnzyme is present and FlangEnzyme is asked for. (rust-lang#963)
1 parent d54daa7 commit 9d58f89

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

enzyme/CMakeLists.txt

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ set(CMAKE_CXX_STANDARD 17)
2424
set(CMAKE_CXX_STANDARD_REQUIRED ON)
2525

2626
option(ENZYME_CLANG "Build enzyme clang plugin" ON)
27+
option(ENZYME_FLANG "Build enzyme flang symlink" OFF)
2728
option(ENZYME_MLIR "Build enzyme mlir plugin" OFF)
2829
option(ENZYME_EXTERNAL_SHARED_LIB "Build external shared library" OFF)
2930
set(ENZYME_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR})
@@ -201,13 +202,13 @@ message("LLVM_INCLUDE_DIRS: ${LLVM_INCLUDE_DIRS}")
201202
message("found llvm definitions " ${LLVM_DEFINITIONS})
202203
message("found llvm version " ${LLVM_VERSION_MAJOR})
203204

204-
option(ENZYME_FLANG "Build for non-version compliant FLANG" OFF)
205-
if (ENZYME_FLANG)
205+
option(ENZYME_FLANG_VERSION "Build for non-version compliant FLANG" OFF)
206+
if (ENZYME_FLANG_VERSION)
206207
add_definitions(-DFLANG=1)
207208
endif()
208209

209-
option(ENZYME_ROCM "Build for non-version compliant ROCM" OFF)
210-
if (ENZYME_ROCM)
210+
option(ENZYME_ROCM_VERSION "Build for non-version compliant ROCM" OFF)
211+
if (ENZYME_ROCM_VERSION)
211212
add_definitions(-DROCM=1)
212213
endif()
213214

@@ -281,6 +282,14 @@ export(TARGETS ClangEnzyme-${LLVM_VERSION_MAJOR}
281282
APPEND FILE "${PROJECT_BINARY_DIR}/EnzymeTargets.cmake")
282283
endif()
283284

285+
if (${ENZYME_FLANG} AND ${Clang_FOUND})
286+
add_custom_target(link_target ALL
287+
COMMAND ${CMAKE_COMMAND} -E create_symlink
288+
ClangEnzyme-${LLVM_VERSION_MAJOR}${LLVM_SHLIBEXT}
289+
${PROJECT_BINARY_DIR}/Enzyme/FlangEnzyme-${LLVM_VERSION_MAJOR}${LLVM_SHLIBEXT}
290+
)
291+
endif()
292+
284293
export(TARGETS LLDEnzyme-${LLVM_VERSION_MAJOR}
285294
APPEND FILE "${PROJECT_BINARY_DIR}/EnzymeTargets.cmake")
286295

0 commit comments

Comments
 (0)