35 lines
900 B
Bash
Executable File
35 lines
900 B
Bash
Executable File
#!/bin/bash
|
|
set -eux -o pipefail
|
|
|
|
source "/Users/distiller/project/env"
|
|
export "PATH=$workdir/miniconda/bin:$PATH"
|
|
pkg="$workdir/final_pkgs/$(ls $workdir/final_pkgs)"
|
|
|
|
# Create a new test env
|
|
# TODO cut all this out into a separate test job and have an entirely different
|
|
# miniconda
|
|
if [[ "$PACKAGE_TYPE" != libtorch ]]; then
|
|
source deactivate || true
|
|
conda create -qyn test python="$DESIRED_PYTHON"
|
|
source activate test >/dev/null
|
|
fi
|
|
|
|
# Install the package
|
|
if [[ "$PACKAGE_TYPE" == libtorch ]]; then
|
|
pkg="$(ls $workdir/final_pkgs/*-latest.zip)"
|
|
unzip "$pkg" -d /tmp
|
|
cd /tmp/libtorch
|
|
elif [[ "$PACKAGE_TYPE" == conda ]]; then
|
|
conda install -y "$pkg"
|
|
else
|
|
pip install "$pkg" -v
|
|
fi
|
|
|
|
# Test
|
|
if [[ "$PACKAGE_TYPE" == libtorch ]]; then
|
|
$workdir/builder/check_binary.sh
|
|
else
|
|
pushd "$workdir/pytorch"
|
|
$workdir/builder/run_tests.sh "$PACKAGE_TYPE" "$DESIRED_PYTHON" "$DESIRED_CUDA"
|
|
fi
|