28 lines
655 B
Bash
28 lines
655 B
Bash
#!/usr/bin/env bash
|
|
# Script used only in CD pipeline
|
|
|
|
set -eou pipefail
|
|
|
|
function do_install() {
|
|
cuda_version=$1
|
|
cuda_version_nodot=${1/./}
|
|
|
|
MAGMA_VERSION="2.6.1"
|
|
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
|
|
|
|
cuda_dir="/usr/local/cuda-${cuda_version}"
|
|
(
|
|
set -x
|
|
tmp_dir=$(mktemp -d)
|
|
pushd ${tmp_dir}
|
|
curl -OLs https://ossci-linux.s3.us-east-1.amazonaws.com/${magma_archive}
|
|
tar -xvf "${magma_archive}"
|
|
mkdir -p "${cuda_dir}/magma"
|
|
mv include "${cuda_dir}/magma/include"
|
|
mv lib "${cuda_dir}/magma/lib"
|
|
popd
|
|
)
|
|
}
|
|
|
|
do_install $1
|