2024-12-12 nightly release (1d4ce73c57417f1af8278af56631de3c25e3bbaf) #1680
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Unit-tests on Linux CPU | |
on: | |
pull_request: | |
push: | |
branches: | |
- nightly | |
- main | |
- release/* | |
workflow_dispatch: | |
env: | |
CHANNEL: "nightly" | |
jobs: | |
tests: | |
strategy: | |
matrix: | |
python_version: ["3.8", "3.9", "3.10"] | |
fail-fast: false | |
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main | |
with: | |
runner: linux.12xlarge | |
repository: pytorch/text | |
script: | | |
# Mark Build Directory Safe | |
git config --global --add safe.directory /__w/text/text | |
# Set up Environment Variables | |
export PYTHON_VERSION="${{ matrix.python_version }}" | |
export VERSION="cpu" | |
export CUDATOOLKIT="cpuonly" | |
# Set CHANNEL | |
if [[ (${GITHUB_EVENT_NAME} = 'pull_request' && (${GITHUB_BASE_REF} = 'release'*)) || (${GITHUB_REF} = 'refs/heads/release'*) ]]; then | |
export CHANNEL=test | |
else | |
export CHANNEL=nightly | |
fi | |
# Create Conda Env | |
conda create -yp ci_env python="${PYTHON_VERSION}" | |
conda activate /work/ci_env | |
python3 -m pip --quiet install cmake>=3.18.0 ninja | |
conda env update --file ".circleci/unittest/linux/scripts/environment.yml" --prune | |
# TorchText-specific Setup | |
printf "* Downloading SpaCy English models\n" | |
python -m spacy download en_core_web_sm | |
printf "* Downloading SpaCy German models\n" | |
python -m spacy download de_core_news_sm | |
# Install PyTorch, Torchvision | |
set -ex | |
conda install \ | |
--yes \ | |
-c "pytorch-${CHANNEL}" \ | |
-c nvidia "pytorch-${CHANNEL}"::pytorch[build="*${VERSION}*"] \ | |
"${CUDATOOLKIT}" | |
python3 setup.py develop | |
python3 -m pip install parameterized | |
# Run Tests | |
python3 -m torch.utils.collect_env | |
cd test | |
python3 -m pytest --cov=torchtext --junitxml=test-results/junit.xml -v --durations 20 torchtext_unittest |