manywheel: Switch DESIRED_CUDA since other scripts rely on it (#626)

Signed-off-by: Eli Uriegas <[email protected]>

Author: seemethere

manywheel/build_rocm.sh

@@ -25,11 +25,10 @@ fi
 # NOTE: We should first check `DESIRED_CUDA` when determining `ROCM_VERSION`
 if [[ -n "$DESIRED_CUDA" ]]; then
     if ! echo "${DESIRED_CUDA}"| grep "^rocm" >/dev/null 2>/dev/null; then
-        ROCM_VERSION="rocm${DESIRED_CUDA}"
-    else
-        # rocm3.7, rocm3.5.1
-        ROCM_VERSION="$DESIRED_CUDA"
+        export DESIRED_CUDA="rocm${DESIRED_CUDA}"
     fi
+    # rocm3.7, rocm3.5.1
+    ROCM_VERSION="$DESIRED_CUDA"
     echo "Using $ROCM_VERSION as determined by DESIRED_CUDA"
 else
     echo "Must set DESIRED_CUDA"