Skip to content

Commit

Permalink
Changing to -eq 1
Browse files Browse the repository at this point in the history
  • Loading branch information
ChSonnabend committed Nov 22, 2024
1 parent 9613081 commit f28a341
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions onnxruntime.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,26 +21,26 @@ mkdir -p $INSTALLROOT

# Check ROCm build conditions

if [ "$ALIBUILD_O2_FORCE_GPU" -ne 0 ] || [ "$ALIBUILD_ENABLE_HIP" -ne 0 ] || \
( ( [ -z "$DISABLE_GPU" ] || [ "$DISABLE_GPU" -eq 0 ] ) && \
( command -v /opt/rocm/bin/rocminfo >/dev/null 2>&1 ) && \
[ -d /opt/rocm/include/hiprand ] && \
[ -d /opt/rocm/include/hipblas ] && \
[ -d /opt/rocm/include/hipsparse ] && \
[ -d /opt/rocm/include/hipfft ] && \
[ -d /opt/rocm/include/rocblas ] && \
[ -d /opt/rocm/include/rocrand ] && \
[ -d /opt/rocm/include/miopen ] && \
[ -d /opt/rocm/include/rccl ] && \
[ -z "$ORT_ROCM_BUILD" ] ); then
if [ "$ALIBUILD_O2_FORCE_GPU" -eq 1 ] || [ "$ALIBUILD_ENABLE_HIP" -eq 1 ] || \
( ( [ -z "$DISABLE_GPU" ] || [ "$DISABLE_GPU" -eq 0 ] ) && \
( command -v /opt/rocm/bin/rocminfo >/dev/null 2>&1 ) && \
[ -d /opt/rocm/include/hiprand ] && \
[ -d /opt/rocm/include/hipblas ] && \
[ -d /opt/rocm/include/hipsparse ] && \
[ -d /opt/rocm/include/hipfft ] && \
[ -d /opt/rocm/include/rocblas ] && \
[ -d /opt/rocm/include/rocrand ] && \
[ -d /opt/rocm/include/miopen ] && \
[ -d /opt/rocm/include/rccl ] && \
[ -z "$ORT_ROCM_BUILD" ] ); then
export ORT_ROCM_BUILD=1
: ${ALIBUILD_O2_OVERRIDE_HIP_ARCHS:="gfx906,gfx908"}
: ${ALIBUILD_O2_OVERRIDE_HIP_ARCHS:="gfx906,gfx908"}
else
export ORT_ROCM_BUILD=0
fi

# Check CUDA build conditions
if [ "$ALIBUILD_O2_FORCE_GPU" -ne 0 ] || [ "$ALIBUILD_ENABLE_CUDA" -ne 0 ] || \
if [ "$ALIBUILD_O2_FORCE_GPU" -eq 1 ] || [ "$ALIBUILD_ENABLE_CUDA" -eq 1 ] || \
( ( [ -z "$DISABLE_GPU" ] || [ "$DISABLE_GPU" -eq 0 ] ) && \
( command -v nvcc >/dev/null 2>&1 ) && \
[ -f /usr/include/cudnn.h ] && \
Expand Down

0 comments on commit f28a341

Please sign in to comment.