Merge pull request #7278 from fwyzard/IB/CMSSW_12_1_X/master_enable_c…

…uDNN_for_ONNX_on_aarch64 Enable the use of cuDNN in ONNX on ARM for CentOS 8
cms-sw · Sep 9, 2021 · 49636ca · 49636ca
2 parents c36813e + dca0e21
commit 49636ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scram-tools.file/tools/onnxruntime/onnxruntime.xml b/scram-tools.file/tools/onnxruntime/onnxruntime.xml
@@ -6,7 +6,7 @@
     <environment name="LIBDIR" default="$ONNXRUNTIME_BASE/lib"/>
   </client>
   <use name="protobuf"/>
-  <ifarchitecture name="!aarch64">
+  <ifarchitecture name="!slc7_aarch64">
     <use name="cuda"/>
     <use name="cudnn"/>
   </ifarchitecture>