Skip to content

Commit

Permalink
[CI] Add another workflow for GraphBolt (#1013)
Browse files Browse the repository at this point in the history
*Issue #, if available:*

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
  • Loading branch information
jalencato authored Sep 7, 2024
1 parent d1b2bff commit 137d481
Show file tree
Hide file tree
Showing 3 changed files with 49 additions and 1 deletion.
2 changes: 2 additions & 0 deletions .github/workflow_scripts/e2e_gb_check.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
#!/bin/bash
python3 -c "import dgl; print(dgl.__version__)"
44 changes: 43 additions & 1 deletion .github/workflows/continuous-integration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ env:
COMMAND-E2E: chmod +x ./e2e_check.sh && ./e2e_check.sh
COMMAND-E2E-MGPU: chmod +x ./e2e_mgpu_check.sh && ./e2e_mgpu_check.sh
COMMAND-E2E-MGPU-MULTITASK: chmod +x ./e2e_mgpu_multitask_check.sh && ./e2e_mgpu_multitask_check.sh
COMMAND-E2E-GB: chmod +x ./e2e_gb_check.sh && ./e2e_gb_check.sh

jobs:

Expand Down Expand Up @@ -310,4 +311,45 @@ jobs:
--remote https://github.com/'${{ github.event.pull_request.head.repo.full_name }}' \
--source-ref '${{ github.event.pull_request.head.sha }}' \
--wait
e2e_graphbolt:
if: ${{ github.event.label.name != 'draft' }}
needs: [pytest_check, lint_check]
runs-on: ubuntu-latest
steps:
- name: Configure AWS Credentials
uses: aws-actions/configure-aws-credentials@v1
with:
role-to-assume: arn:aws:iam::698571788627:role/github-oidc-role
aws-region: us-east-1
- name: Checkout repository
uses: actions/checkout@v3
- name: Set up Python 3.x
uses: actions/setup-python@v4
with:
python-version: '3.x'
- name: Install dependencies
shell: bash
run: |
python3 -m pip install --upgrade pip
python3 -m pip install pytest
pip3 install boto3
- name: Submit Job (for Push)
if: ${{ github.event_name == 'push' }}
shell: bash
run: |
echo "Start submitting job - Push"
python3 ./submitJob.py --job-type CI-GPU-DGL230 --name Graphstorm-E2E-Test-GB-Push-'${{ github.ref }}' \
--command "${{ env.COMMAND-E2E-GB }}" \
--remote https://github.com/'${{ github.repository }}' \
--source-ref '${{ github.ref }}' \
--wait
- name: Submit Job (for Pull Request)
if: ${{ github.event_name == 'pull_request_target' }}
shell: bash
run: |
echo "Start submitting job - Pull Request"
python3 ./submitJob.py --job-type CI-GPU-DGL230 --name Graphstorm-E2E-Test-GB-PR#-'${{ github.event.number }}' \
--command "${{ env.COMMAND-E2E-GB }}" \
--remote https://github.com/'${{ github.event.pull_request.head.repo.full_name }}' \
--source-ref '${{ github.event.pull_request.head.sha }}' \
--wait
4 changes: 4 additions & 0 deletions submitJob.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,10 @@
'job_definition': 'graphstorm-definition-gpu',
'job_queue': 'graphstorm-queue-v1',
},
'CI-GPU-DGL230': {
'job_definition': 'graphstorm-definition-gpu-dgl230',
'job_queue': 'graphstorm-queue-graphbolt',
}
'CI-LINT': {
'job_definition': 'graphstorm-definition-lint',
'job_queue': 'graphstorm-queue-lint',
Expand Down

0 comments on commit 137d481

Please sign in to comment.