cavis/.github/workflows/run-gpu-tests-sanity-checks...

29 lines
980 B
YAML
Raw Normal View History

2021-03-10 00:58:56 +01:00
on:
push:
jobs:
linux-x86_64-cuda-11-2:
runs-on: [self-hosted]
steps:
- uses: AutoModality/action-clean@v1
- name: Cancel Previous Runs
uses: styfle/cancel-workflow-action@0.8.0
with:
access_token: ${{ github.token }}
2021-03-10 01:08:14 +01:00
- uses: ./.github/actions/download-dl4j-test-resources-linux
2021-03-10 00:58:56 +01:00
- uses: actions/checkout@v2
- name: Run cuda tests
shell: bash
env:
DEBIAN_FRONTEND: noninteractive
run: |
export PATH="/opt/protobuf/bin:/usr/local/cuda-11.2/bin:$PATH"
nvcc --version
mvn --version
cmake --version
protoc --version
bash ./change-cuda-versions.sh 11.2
export OMP_NUM_THREADS=1
2021-03-10 01:04:48 +01:00
mvn -Ptestresources -pl ":deeplearning4j-modelimport,:deeplearning4j-core,:nd4j-cuda-11.2,:samediff-import,:libnd4j" -Dlibnd4j.compute="5.0 5.2 5.3 6.0 6.2 8.0" -Ptest-nd4j-cuda --also-make -Dlibnd4j.chip=cuda clean test
2021-03-10 00:58:56 +01:00