Skip to content

Commit

Permalink
Merge pull request #7409 from ddaina/disable_cudnn_slc7_aarch64
Browse files Browse the repository at this point in the history
don't build cudnn on slc7_aarch64
  • Loading branch information
smuzaffar authored Oct 26, 2021
2 parents 5a3fb7e + 68968d0 commit a38296e
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions onnxruntime.spec
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,9 @@ Source1: https://github.com/microsoft/onnxruntime/commit/de4089f8cbe0baffe56a363

BuildRequires: cmake ninja
Requires: protobuf py3-numpy py3-wheel py3-onnx zlib libpng py3-pybind11
%if "%{cmsos}" != "slc7_aarch64"
Requires: cuda cudnn
%endif

%prep
%setup -q -n %{n}-%{realversion}
Expand All @@ -18,11 +20,11 @@ patch -p1 < %{_sourcedir}/de4089f8cbe0baffe56a363cc3a41595cc8f0809.patch
%build
rm -rf ../build; mkdir ../build; cd ../build

if [ $(%{cuda_gcc_support}) = true ]; then
%if "%{cmsos}" != "slc7_aarch64"
USE_CUDA=ON
else
%else
USE_CUDA=OFF
fi
%endif

cmake ../%{n}-%{realversion}/cmake -GNinja \
-DPYTHON_EXECUTABLE=${PYTHON3_ROOT}/bin/python3 \
Expand Down

0 comments on commit a38296e

Please sign in to comment.