Add RAPIDS_ARCH to override host arch (#10)

* Add RAPIDS_PY_WHEEL_ARCH to override host arch * Override arch with RAPIDS_ARCH
rapidsai · Sep 22, 2022 · 7632753 · 7632753
1 parent f444b84
commit 7632753
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tools/rapids-package-name b/tools/rapids-package-name
@@ -56,7 +56,13 @@ if (( append_pyver == 1 )); then
 fi
 
 # for cpp and python package types, always append arch
-pkg_name+="_$(arch)"
+if [[ -v RAPIDS_ARCH ]] && [[ "${RAPIDS_ARCH}" != "" ]]; then
+  # use arch override if specified
+  pkg_name+="_${RAPIDS_ARCH}"
+else
+  # otherwise use architecture of the host that's running the upload command
+  pkg_name+="_$(arch)"
+fi
 
 # for cpp and python package types, its a tarball, append .tar.gz and prepend project name
 pkg_name="${repo_name}_${pkg_name}.tar.gz"