#!/bin/bash

if [[ "$BUILD_ENVIRONMENT" == "pytorch-linux-xenial-py3-clang5-asan" ]]; then
  exec "$(dirname "${BASH_SOURCE[0]}")/build-asan.sh" $*
fi

# Add nccl2 for distributed test.
apt-get install libnccl-dev libnccl2

# Required environment variable: $BUILD_ENVIRONMENT
# (This is set by default in the Docker images we build, so you don't
# need to set it yourself.

COMPACT_JOB_NAME="${BUILD_ENVIRONMENT}-build"
source "$(dirname "${BASH_SOURCE[0]}")/common.sh"

echo "Python version:"
python --version

echo "GCC version:"
gcc --version

# TODO: Don't run this...
pip install -r requirements.txt || true

if ! which conda; then
  pip install mkl mkl-devel
fi

# Test ATen
if [[ "$BUILD_ENVIRONMENT" != *cuda* ]]; then
  echo "Testing ATen"
  time NO_CUDA=1 tools/run_aten_tests.sh
fi