mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-18 03:34:58 +00:00
74 lines
2.9 KiB
Diff
74 lines
2.9 KiB
Diff
diff --git a/hip-config.cmake.in b/hip-config.cmake.in
|
|
index 89d1224e..120b68c6 100755
|
|
--- a/hip-config.cmake.in
|
|
+++ b/hip-config.cmake.in
|
|
@@ -142,7 +142,7 @@ if(HIP_COMPILER STREQUAL "clang")
|
|
file(TO_CMAKE_PATH "${HIP_PATH}/../lc" HIP_CLANG_ROOT)
|
|
endif()
|
|
else()
|
|
- set(HIP_CLANG_ROOT "${ROCM_PATH}/llvm")
|
|
+ set(HIP_CLANG_ROOT "@clang@")
|
|
endif()
|
|
if(NOT HIP_CXX_COMPILER)
|
|
set(HIP_CXX_COMPILER ${CMAKE_CXX_COMPILER})
|
|
@@ -206,10 +206,7 @@ if(NOT WIN32)
|
|
#if HSA is not under ROCm then provide CMAKE_PREFIX_PATH=<HSA_PATH>
|
|
find_path(HSA_HEADER hsa/hsa.h
|
|
PATHS
|
|
- "${_IMPORT_PREFIX}/include"
|
|
- #FILE_REORG_BACKWARD_COMPATIBILITY ${_IMPORT_PREFIX}/../include is for Backward compatibility
|
|
- "${_IMPORT_PREFIX}/../include"
|
|
- ${ROCM_PATH}/include
|
|
+ "@rocm_runtime@/include"
|
|
)
|
|
|
|
if (NOT HSA_HEADER)
|
|
@@ -224,8 +221,8 @@ set_target_properties(hip::host PROPERTIES
|
|
|
|
if(HIP_RUNTIME MATCHES "rocclr")
|
|
set_target_properties(hip::amdhip64 PROPERTIES
|
|
- INTERFACE_INCLUDE_DIRECTORIES "${_IMPORT_PREFIX}/include;${HSA_HEADER}"
|
|
- INTERFACE_SYSTEM_INCLUDE_DIRECTORIES "${_IMPORT_PREFIX}/include;${HSA_HEADER}"
|
|
+ INTERFACE_INCLUDE_DIRECTORIES "@hip@/include;${HSA_HEADER}"
|
|
+ INTERFACE_SYSTEM_INCLUDE_DIRECTORIES "@hip@/include;${HSA_HEADER}"
|
|
)
|
|
|
|
get_target_property(amdhip64_type hip::amdhip64 TYPE)
|
|
@@ -233,8 +230,8 @@ if(HIP_RUNTIME MATCHES "rocclr")
|
|
|
|
if(NOT WIN32)
|
|
set_target_properties(hip::device PROPERTIES
|
|
- INTERFACE_INCLUDE_DIRECTORIES "${_IMPORT_PREFIX}/include"
|
|
- INTERFACE_SYSTEM_INCLUDE_DIRECTORIES "${_IMPORT_PREFIX}/include"
|
|
+ INTERFACE_INCLUDE_DIRECTORIES "@hip@/include"
|
|
+ INTERFACE_SYSTEM_INCLUDE_DIRECTORIES "@hip@/include"
|
|
)
|
|
endif()
|
|
endif()
|
|
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
|
|
index 83866d83..4125d3aa 100644
|
|
--- a/src/CMakeLists.txt
|
|
+++ b/src/CMakeLists.txt
|
|
@@ -178,7 +178,7 @@ if(__HIP_ENABLE_PCH)
|
|
${ROCM_PATH}/llvm)
|
|
# find_package(LLVM) returns the lib/cmake/llvm location. We require the root.
|
|
if(NOT DEFINED HIP_LLVM_ROOT)
|
|
- set(HIP_LLVM_ROOT "${LLVM_DIR}/../../..")
|
|
+ set(HIP_LLVM_ROOT "@clang@")
|
|
endif()
|
|
|
|
execute_process(COMMAND sh -c "${CMAKE_CURRENT_SOURCE_DIR}/hip_embed_pch.sh ${HIP_COMMON_INCLUDE_DIR} ${PROJECT_BINARY_DIR}/include ${PROJECT_SOURCE_DIR}/include ${HIP_LLVM_ROOT}" COMMAND_ECHO STDERR RESULT_VARIABLE EMBED_PCH_RC)
|
|
diff --git a/src/hip_embed_pch.sh b/src/hip_embed_pch.sh
|
|
index 0a1572b2..aa855d63 100755
|
|
--- a/src/hip_embed_pch.sh
|
|
+++ b/src/hip_embed_pch.sh
|
|
@@ -149,7 +149,7 @@ EOF
|
|
|
|
$LLVM_DIR/bin/clang -cc1 -O3 -emit-pch -triple amdgcn-amd-amdhsa -aux-triple x86_64-unknown-linux-gnu -fcuda-is-device -std=c++17 -fgnuc-version=4.2.1 -o $tmp/hip_wave64.pch -x hip-cpp-output - <$tmp/pch_wave64.cui &&
|
|
|
|
- $LLVM_DIR/bin/llvm-mc -o hip_pch.o $tmp/hip_pch.mcin --filetype=obj &&
|
|
+ @llvm@/bin/llvm-mc -o hip_pch.o $tmp/hip_pch.mcin --filetype=obj &&
|
|
|
|
rm -rf $tmp
|
|
}
|