Workflow fixes (#11270)

This commit is contained in:
Lysandre Debut
2021-04-15 23:21:17 -04:00
committed by GitHub
parent dfc6dd8584
commit 5254220e7f
3 changed files with 46 additions and 2 deletions

View File

@@ -38,6 +38,7 @@ jobs:
apt -y update && apt install -y libsndfile1-dev
pip install --upgrade pip
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
- name: Are GPUs recognized by our DL frameworks
run: |
@@ -121,6 +122,7 @@ jobs:
apt -y update && apt install -y libsndfile1-dev
pip install --upgrade pip
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
- name: Are GPUs recognized by our DL frameworks
run: |
@@ -220,6 +222,13 @@ jobs:
if: ${{ always() }}
run: cat reports/tests_torch_cuda_extensions_gpu_failures_short.txt
- name: Test suite reports artifacts
if: ${{ always() }}
uses: actions/upload-artifact@v2
with:
name: run_tests_torch_cuda_extensions_gpu_test_reports
path: reports
run_tests_torch_cuda_extensions_multi_gpu:
runs-on: [self-hosted, docker-gpu, multi-gpu]
container:
@@ -253,6 +262,13 @@ jobs:
if: ${{ always() }}
run: cat reports/tests_torch_cuda_extensions_multi_gpu_failures_short.txt
- name: Test suite reports artifacts
if: ${{ always() }}
uses: actions/upload-artifact@v2
with:
name: run_tests_torch_cuda_extensions_multi_gpu_test_reports
path: reports
send_results:
name: Send results to webhook