diff --git a/.github/workflows/test-runner.yml b/.github/workflows/test-runner.yml index 12c32b828..ad5415a8c 100644 --- a/.github/workflows/test-runner.yml +++ b/.github/workflows/test-runner.yml @@ -26,7 +26,7 @@ on: gpu_type: type: string default: "" - description: "GPU type for CUDA testing: T4, L40S" + description: "GPU type for CUDA testing: T4, A10,L40S" # cpu_type currently only affects linux x64 CPU testing to select specific CPU architectures cpu_type: type: string @@ -65,11 +65,14 @@ jobs: T4) TEST_RUNNER="bandb-aws-g4dn-4xlarge-plus-use1-public-80" ;; + A10) + TEST_RUNNER="bandb-aws-g5-4xlarge-plus-use1-public-80" + ;; L40S) TEST_RUNNER="bandb-aws-g6e-4xlarge-plus-use1-public-80" ;; *) - echo "::error::Must specify gpu_type (T4 or L40S) for linux-x64 cuda backend" + echo "::error::Must specify gpu_type (T4, A10, L40S) for linux-x64 cuda backend" exit 1 ;; esac @@ -164,7 +167,7 @@ jobs: run: bash .github/scripts/build-cuda.sh env: cuda_version: ${{ inputs.cuda_version }} - cuda_targets: "75;89" + cuda_targets: "75;80;89" - name: Upload build artifact uses: actions/upload-artifact@v4 diff --git a/.github/workflows/tests-nightly.yml b/.github/workflows/tests-nightly.yml index 60ed4fe3f..754b0cd36 100644 --- a/.github/workflows/tests-nightly.yml +++ b/.github/workflows/tests-nightly.yml @@ -56,7 +56,7 @@ jobs: matrix: # Linux x64 cross-product platform: [linux-x64] - gpu_type: [T4, L40S] + gpu_type: [T4, A10, L40S] cuda_version: ["11.8.0", "12.6.3", "12.8.1", "13.0.2"] include: diff --git a/.github/workflows/tests-pr.yml b/.github/workflows/tests-pr.yml index 7766089f6..ade85bcea 100644 --- a/.github/workflows/tests-pr.yml +++ b/.github/workflows/tests-pr.yml @@ -64,7 +64,7 @@ jobs: fail-fast: false matrix: platform: [linux-x64] - gpu_type: [T4, L40S] + gpu_type: [T4, A10, L40S] cuda_version: ["11.8.0", "12.8.1", "13.0.2"] include: