mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
32099961d5
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160479 Approved by: https://github.com/izaitsevfb ghstack dependencies: #160477
25 lines
536 B
Bash
Executable File
25 lines
536 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -ex
|
|
|
|
SCRIPTPATH="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
|
|
|
|
case "${GPU_ARCH_TYPE:-BLANK}" in
|
|
cuda)
|
|
bash "${SCRIPTPATH}/build_cuda.sh"
|
|
;;
|
|
rocm)
|
|
bash "${SCRIPTPATH}/build_rocm.sh"
|
|
;;
|
|
cpu | cpu-cxx11-abi | cpu-s390x)
|
|
bash "${SCRIPTPATH}/build_cpu.sh"
|
|
;;
|
|
xpu)
|
|
bash "${SCRIPTPATH}/build_xpu.sh"
|
|
;;
|
|
*)
|
|
echo "Un-recognized GPU_ARCH_TYPE '${GPU_ARCH_TYPE}', exiting..."
|
|
exit 1
|
|
;;
|
|
esac
|