Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
deepspeed
Commits
7833aed2
Unverified
Commit
7833aed2
authored
Jan 27, 2021
by
Jeff Rasley
Committed by
GitHub
Jan 27, 2021
Browse files
Create torch16.yml (#699)
parent
5221832e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
44 additions
and
0 deletions
+44
-0
.github/workflows/torch16.yml
.github/workflows/torch16.yml
+44
-0
No files found.
.github/workflows/torch16.yml
0 → 100644
View file @
7833aed2
# Unit test config for manual use on torch1.6 runners
name
:
Torch16
# Controls when the action will run.
on
:
pull_request
:
paths-ignore
:
-
'
docs/**'
# Allows you to run this workflow manually from the Actions tab
workflow_dispatch
:
# 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
,
torch1.6
]
# 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
# 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/
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment