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
"...git@developer.sourcefind.cn:modelzoo/qwen_lmdeploy.git" did not exist on "b239346701bd8d9cbc2ba1a2f5053cb1e1d671b5"
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