on: workflow_dispatch: jobs: # Wait for up to a minute for previous run to complete, abort if not done by then pre-ci: runs-on: ubuntu-latest timeout-minutes: 1 steps: - name: 'Block Concurrent Executions' uses: softprops/turnstyle@v1 with: poll-interval-seconds: 10 env: GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} linux-x86_64-cuda-11-2: needs: pre-ci 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 }} - uses: ./.github/actions/download-dl4j-test-resources-linux - 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 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