diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 28c7bdc..b43e7fd 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -1,65 +1,15 @@ -name: Merge group tests +name: GPU CI on: - merge_group: - pull_request: + push: + branches: + - gpu-ci jobs: - test-ref: - runs-on: ubuntu-latest - if: github.event_name == 'merge_group' + gpu-ci: + runs-on: gpu-ci steps: - uses: actions/checkout@v4 - - name: Print branch names + - name: Test GPU run: | - echo "Head ref: ${{ github.event.merge_group.head_ref }}" - echo "Base ref: ${{ github.event.merge_group.base_ref }}" - - uses: actions/checkout@v4 - with: - ref: ${{ github.event.merge_group.base_ref }} - - test-ref-b: - runs-on: ubuntu-latest - if: github.event_name == 'pull_request' - steps: - - uses: actions/checkout@v4 - - name: Print branch name - run: | - echo "Base ref: ${{ github.base_ref }}" - - check-job-results: - if: always() - needs: [test-ref, test-ref-b] - runs-on: ubuntu-latest - steps: - - name: Check job results - id: check-results - run: | - # Create an associative array of job statuses - declare -A job_results=( - ["test-ref"]="${{ needs.test-ref.result }}" - ["test-ref-b"]="${{ needs.test-ref-b.result }}" - ) - - # Iterate through jobs and get their results - failed_count=0 - for job in "${!job_results[@]}"; do - RESULT=${job_results[$job]} - if [[ "$RESULT" == "failure" ]]; then - failed_count=$((failed_count + 1)) - fi - echo "$job result: $RESULT" - done - - if [ "$failed_count" -gt 0 ]; then - echo "Some jobs failed" - echo "result=failure" | tee -a $GITHUB_OUTPUT - else - echo "All jobs succeeded or were skipped" - echo "result=success" | tee -a $GITHUB_OUTPUT - fi - - name: Error on job failure - run: | - if [[ "${{ steps.check-results.outputs.result }}" == "failure" ]]; then - exit 1 - fi + nvidia-smi