comp-chem/cp2k
1
0
Fork 0
mirror of https://github.com/cp2k/cp2k.git synced 2026-07-29 06:35:28 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Toolchain: Hardcode HIP_PLATFORM=nvidia for now

Browse source
This commit is contained in:
Ole Schütt 2021-06-16 11:54:57 +02:00 • committed by Ole Schütt
parent ba4c83a26a
commit 3c81dfb148
1 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tools/toolchain/scripts/generate_arch_files.sh
View file

@ -140,22 +140,37 @@ fi
# HIP handling
if [ "${ENABLE_HIP}" = __TRUE__ ] && [ "${GPUVER}" != no ]; then
LIBS+=" IF_HIP(-lhipblas|)" # -lamdhip64
HIPFLAGS="-I\\\${ROCM_PATH}/hip/include -I\\\${ROCM_PATH}/hipblas/include"
HIPFLAGS+=" -g -arch sm_${ARCH_NUM} -O3 -Xcompiler='-fopenmp' --std=c++11 \$(DFLAGS)"
#
# TODO: Make toolchain oblivious to HIP_PLATFORM, currently it only works for Nvidia.
#
# DBCSR suffers from https://github.com/ROCm-Developer-Tools/HIP/issues/268
DFLAGS+=" IF_HIP(-D__HIP_PLATFORM_NVIDIA__ -D__GRID_HIP|)" # -D__DBCSR_ACC
HIP_FLAGS="-g -arch sm_${ARCH_NUM} -O3 -Xcompiler='-fopenmp' --std=c++11 \$(DFLAGS)"
LIBS+=" IF_HIP(-lhipblas -lcudart -lnvrtc -lcuda -lcufft -lcublas -lrt|)" # -lamdhip64
check_command hipcc "hip"
#check_lib -lamdhip64 "hip"
check_lib -lhipblas "hip"
# DBCSR suffers from https://github.com/ROCm-Developer-Tools/HIP/issues/268
#DFLAGS+=" IF_HIP(-D__DBCSR_ACC|)"
DFLAGS+=" IF_HIP(-D__GRID_HIP|)"
check_command nvcc "cuda"
check_lib -lcudart "cuda"
check_lib -lnvrtc "cuda"
check_lib -lcuda "cuda"
check_lib -lcufft "cuda"
check_lib -lcublas "cuda"
# Set LD-flags
HIP_LDFLAGS=''
#add_lib_from_paths HIP_LDFLAGS "libamdhip64.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libhipblas.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libcudart.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libnvrtc.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libcuda.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libcufft.*" $LIB_PATHS
add_lib_from_paths HIP_LDFLAGS "libcublas.*" $LIB_PATHS
LDFLAGS+=" ${HIP_LDFLAGS}"
HIP_INCLUDES="-I\${ROCM_PATH}/hip/include -I\${ROCM_PATH}/hipblas/include -I\${CUDA_PATH}/include"
fi
# -------------------------
@ -199,11 +214,11 @@ EOF
cat << EOF >> $__filename
#
CXX = \${CC}
CXXFLAGS = \${CXXFLAGS} -std=c++11 -fopenmp
CFLAGS = \${CFLAGS}
CXXFLAGS = \${CXXFLAGS} \${HIP_INCLUDES} -std=c++11 -fopenmp
CFLAGS = \${CFLAGS} \${HIP_INCLUDES}
GPUVER = \${GPUVER}
OFFLOAD_CC = \${ROCM_PATH}/hip/bin/hipcc
OFFLOAD_FLAGS = \${HIPFLAGS}
OFFLOAD_FLAGS = \${HIP_FLAGS} \${HIP_INCLUDES}
EOF
fi