@@ -287,13 +287,6 @@ if [ ! $HIPFORT_GPU ] ; then
287
287
fi
288
288
fi
289
289
290
- # Future versions of ROCm will install the device library in ROCM_PATH/amdgcn/bitcode
291
- if [ -d $ROCM_PATH /amdgcn/bitcode ] ; then
292
- DEVICE_LIB_PATH=$ROCM_PATH /amdgcn/bitcode
293
- else
294
- DEVICE_LIB_PATH=$ROCM_PATH /lib
295
- fi
296
-
297
290
# Handle all differences between amdgcn and nvidia here
298
291
if [ " ${HIPFORT_GPU: 0: 3} " == " sm_" ] ; then
299
292
TARGET_ARCH=" nvptx"
@@ -319,7 +312,18 @@ else
319
312
TARGET_TRIPLE=${TARGET_TRIPLE:- amdgcn-amd-amdhsa}
320
313
TARGET_ARCH=" amdgcn"
321
314
TARGET_LIBS=" -L$ROCM_PATH /lib -lamdhip64 -Wl,-rpath=$ROCM_PATH /lib "
322
- HIPCC_ENV=" HIP_PLATFORM=$HIP_PLATFORM DEVICE_LIB_PATH=$DEVICE_LIB_PATH HIP_CLANG_PATH=$ROCM_PATH /llvm/bin"
315
+ HIPCC_ENV=" HIP_PLATFORM=$HIP_PLATFORM "
316
+ if [ -z ${HIP_CLANG_PATH+x} ]; then
317
+ HIPCC_ENV+=" HIP_CLANG_PATH=$ROCM_PATH /llvm/bin"
318
+ fi
319
+ if [ -z ${DEVICE_LIB_PATH+x} ]; then
320
+ # Future versions of ROCm will install the device library in ROCM_PATH/amdgcn/bitcode
321
+ if [ -d $ROCM_PATH /amdgcn/bitcode ] ; then
322
+ HIPCC_ENV+=" DEVICE_LIB_PATH=$ROCM_PATH /amdgcn/bitcode"
323
+ else
324
+ HIPCC_ENV+=" DEVICE_LIB_PATH=$ROCM_PATH /lib"
325
+ fi
326
+ fi
323
327
HIPCC_OPTS=" -fno-gpu-rdc -fPIC --offload-arch=$HIPFORT_GPU "
324
328
fi
325
329
@@ -472,9 +476,7 @@ if [ "$__HIPCC_INPUTS" != "" ] ; then
472
476
echo " Please install hip"
473
477
exit $DEADRC
474
478
fi
475
- [ $VERBOSE ] && echo " export $HIPCC_ENV "
476
- export $HIPCC_ENV
477
- runcmd " $ROCM_PATH /bin/hipcc $HIPCC_OPTS $PASSTHRUARGS $__HIPCC_INPUTS $__HIPCC_LINKOPTS -o $__HIPCC_OUTFILE "
479
+ runcmd " $HIPCC_ENV $ROCM_PATH /bin/hipcc $HIPCC_OPTS $PASSTHRUARGS $__HIPCC_INPUTS $__HIPCC_LINKOPTS -o $__HIPCC_OUTFILE "
478
480
fi
479
481
480
482
if [ " $__INPUTS " != " " ] ; then
0 commit comments