GPU tests on multiple instance types #1
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: GPU tests on multiple instance types | |
| on: | |
| push: | |
| branches: [main] | |
| pull_request: | |
| branches: [main] | |
| workflow_dispatch: | |
| permissions: | |
| id-token: write | |
| contents: read | |
| jobs: | |
| test-g5: | |
| name: Test on g5.2xlarge (A10G) | |
| uses: ./.github/workflows/test.yaml | |
| with: | |
| instance_type: g5.2xlarge | |
| secrets: inherit | |
| test-g6: | |
| name: Test on g6.2xlarge (L4) | |
| uses: ./.github/workflows/test.yaml | |
| with: | |
| instance_type: g6.2xlarge | |
| secrets: inherit | |
| summary: | |
| name: Test Summary | |
| needs: [test-g5, test-g6] | |
| runs-on: ubuntu-latest | |
| if: always() | |
| steps: | |
| - name: Summary | |
| run: | | |
| echo "## Test Results Summary" >> $GITHUB_STEP_SUMMARY | |
| echo "" >> $GITHUB_STEP_SUMMARY | |
| # Check g5.2xlarge result | |
| if [ "${{ needs.test-g5.result }}" == "success" ]; then | |
| echo "✅ **g5.2xlarge (A10G)**: Tests passed" >> $GITHUB_STEP_SUMMARY | |
| elif [ "${{ needs.test-g5.result }}" == "failure" ]; then | |
| echo "❌ **g5.2xlarge (A10G)**: Tests failed" >> $GITHUB_STEP_SUMMARY | |
| elif [ "${{ needs.test-g5.result }}" == "cancelled" ]; then | |
| echo "⚠️ **g5.2xlarge (A10G)**: Tests cancelled" >> $GITHUB_STEP_SUMMARY | |
| else | |
| echo "⏭️ **g5.2xlarge (A10G)**: Tests skipped" >> $GITHUB_STEP_SUMMARY | |
| fi | |
| # Check g6.2xlarge result | |
| if [ "${{ needs.test-g6.result }}" == "success" ]; then | |
| echo "✅ **g6.2xlarge (L4)**: Tests passed" >> $GITHUB_STEP_SUMMARY | |
| elif [ "${{ needs.test-g6.result }}" == "failure" ]; then | |
| echo "❌ **g6.2xlarge (L4)**: Tests failed" >> $GITHUB_STEP_SUMMARY | |
| elif [ "${{ needs.test-g6.result }}" == "cancelled" ]; then | |
| echo "⚠️ **g6.2xlarge (L4)**: Tests cancelled" >> $GITHUB_STEP_SUMMARY | |
| else | |
| echo "⏭️ **g6.2xlarge (L4)**: Tests skipped" >> $GITHUB_STEP_SUMMARY | |
| fi | |
| echo "" >> $GITHUB_STEP_SUMMARY | |
| echo "### GPU Details" >> $GITHUB_STEP_SUMMARY | |
| echo "- **g5.2xlarge**: NVIDIA A10G (Ampere, compute capability 8.6)" >> $GITHUB_STEP_SUMMARY | |
| echo "- **g6.2xlarge**: NVIDIA L4 (Ada Lovelace, compute capability 8.9)" >> $GITHUB_STEP_SUMMARY | |
| # Set overall status | |
| if [ "${{ needs.test-g5.result }}" == "failure" ] || [ "${{ needs.test-g6.result }}" == "failure" ]; then | |
| echo "::error::One or more test jobs failed" | |
| exit 1 | |
| fi |