Merge branch 'master' into jeffra/inject_v2 #3602
Workflow file for this run
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
# This is a basic workflow to help you get started with Actions | |
name: Build | |
# Controls when the action will run. | |
on: | |
push: | |
paths-ignore: | |
- 'docs/**' | |
pull_request: | |
paths-ignore: | |
- 'docs/**' | |
# A workflow run is made up of one or more jobs that can run sequentially or in parallel | |
jobs: | |
# This workflow contains a single job called "build" | |
build: | |
# The type of runner that the job will run on | |
runs-on: self-hosted | |
# Steps represent a sequence of tasks that will be executed as part of the job | |
steps: | |
# Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it | |
- uses: actions/checkout@v2 | |
# Runs a single command using the runners shell | |
- name: environment | |
run: | | |
nvidia-smi | |
which python | |
python --version | |
which nvcc | |
nvcc --version | |
python -c "import torch; print('torch:', torch.__version__, torch)" | |
python -c "import torch; print('CUDA available:', torch.cuda.is_available())" | |
# Runs a set of commands using the runners shell | |
- name: Install deepspeed | |
run: | | |
pip install .[dev] | |
ds_report | |
- name: Formatting checks | |
run: | | |
pre-commit run --all-files | |
# Runs a set of commands using the runners shell | |
- name: Unit tests | |
run: | | |
if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi | |
TORCH_EXTENSIONS_DIR=./torch-extensions pytest --durations=0 --forked --verbose -x tests/unit/ |