Update cmsCudaSetup.sh handling of CUDA_FLAGS (cms-sw#518)

Support having multiple `-gencode arch=compute_..,code=sm_..` options on a single line, as well as one set of options on each line.
fwyzard · Jul 18, 2020 · 1e0ab2e · 1e0ab2e
1 parent e2840ed
commit 1e0ab2e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/HeterogeneousCore/CUDAServices/scripts/cmsCudaSetup.sh b/HeterogeneousCore/CUDAServices/scripts/cmsCudaSetup.sh
@@ -6,7 +6,8 @@ DOTS=$(cudaComputeCapabilities | awk '{ print $2 }' | sort -u)
 CAPS=$(echo $DOTS | sed -e's#\.*##g')
 
 # remove existing capabilities
-sed -i $TOOL -e'\#<flags CUDA_FLAGS="-gencode arch=compute_..,code=sm_.."/>#d'
+sed -i $TOOL -e"s#-gencode arch=compute_..,code=sm_.. *##g"
+sed -i $TOOL -e"\#<flags CUDA_FLAGS=\"\"/>#d"
 
 # add support for the capabilities found on this machine
 for CAP in $CAPS; do