Skip to content

Commit

Permalink
add deepspeed fairscale (huggingface#10116)
Browse files Browse the repository at this point in the history
  • Loading branch information
stas00 authored Feb 10, 2021
1 parent d478257 commit 937f670
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions .github/workflows/self-scheduled.yml
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ jobs:
- name: Failure short reports
if: ${{ always() }}
run: cat reports/tests_torch_gpu_failures_short.txt

- name: Run examples tests on GPU
if: ${{ always() }}
env:
Expand Down Expand Up @@ -157,7 +157,7 @@ jobs:
run: |
source .env/bin/activate
python -m pytest -n 1 --dist=loadfile -s --make-reports=tests_tf_gpu tests
- name: Failure short reports
if: ${{ always() }}
run: cat reports/tests_tf_gpu_failures_short.txt
Expand All @@ -183,7 +183,7 @@ jobs:
with:
name: run_all_tests_tf_gpu_test_reports
path: reports

run_all_tests_torch_multi_gpu:
runs-on: [self-hosted, gpu, multi-gpu]
steps:
Expand Down Expand Up @@ -221,6 +221,8 @@ jobs:
pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets
pip install fairscale
pip install deepspeed
pip list
- name: Are GPUs recognized by our DL frameworks
Expand Down Expand Up @@ -342,7 +344,7 @@ jobs:
run: |
source .env/bin/activate
python -m pytest -n 1 --dist=loadfile -s -m is_pipeline_test --make-reports=tests_tf_pipeline_multi_gpu tests
- name: Failure short reports
if: ${{ always() }}
run: cat reports/tests_tf_pipeline_multi_gpu_failures_short.txt
Expand All @@ -353,4 +355,3 @@ jobs:
with:
name: run_all_tests_tf_multi_gpu_test_reports
path: reports

0 comments on commit 937f670

Please sign in to comment.