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
0518252d
Commit
0518252d
authored
Dec 11, 2020
by
Jeff Rasley
Browse files
add manual workflow to run tests with precompiled ops
parent
7300f3e3
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
47 additions
and
0 deletions
+47
-0
.github/workflows/pre-compile-ops.yml
.github/workflows/pre-compile-ops.yml
+47
-0
No files found.
.github/workflows/pre-compile-ops.yml
0 → 100644
View file @
0518252d
# This is a basic workflow to help you get started with Actions
name
:
Tests-w-precompiled-ops
# Controls when the action will run.
on
:
# 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
# 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
:
|
DS_BUILD_OPS=1 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/
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