Skip to content

Commit

Permalink
ci: add GPU workflows
Browse files Browse the repository at this point in the history
  • Loading branch information
mspronesti committed Jul 29, 2024
1 parent 1a5b889 commit 4e1c43d
Show file tree
Hide file tree
Showing 2 changed files with 54 additions and 6 deletions.
32 changes: 26 additions & 6 deletions .github/workflows/ci-dev.yml
Original file line number Diff line number Diff line change
@@ -1,14 +1,31 @@
name: oneAPI-build
name: CI kernels

on: [push, pull_request]

jobs:
setup:
gpu:
runs-on: devcloud # self-hosted GPU runner on Intel devcloud
name: Run all kernels (GPU)
steps:
- uses: actions/checkout@v3

- name: Build kernels (GPU)
run: |
cd dev/sycl
make CC=icx
- name: Run kernels (GPU)
run: |
cd dev/sycl
make run_all CC=icx
cpu:
runs-on: ubuntu-latest
name: Run all kernels (CPU)
steps:
- uses: actions/checkout@v3
- uses: rscohn2/setup-oneapi@v0
- name: Install OneAPI
uses: rscohn2/setup-oneapi@v0
with:
list: true
components: |
Expand All @@ -20,15 +37,18 @@ jobs:
ippcp
mkl
tbb
- name: Setup variables
run: |
source /opt/intel/oneapi/setvars.sh
printenv >> $GITHUB_ENV
- name: Build kernels
- name: Build kernels (CPU)
run: |
cd dev/sycl
make CC=icx
- name: Run kernels
- name: Run kernels (CPU)
run: |
cd dev/sycl
make run_all CC=icx
make run_all CC=icx
28 changes: 28 additions & 0 deletions .github/workflows/ci-train-gpu.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
name: GPU Train and Test

on: [push, pull_request]

jobs:
train-and-test-gpu:
runs-on: devcloud # self-hosted GPU runner on Intel devcloud

steps:
- name: Checkout code
uses: actions/checkout@v4

- name: Download starter pack (initial weights)
run: |
chmod u+x ./download_starter_pack.sh
./download_starter_pack.sh
- name: Compile training and testing program
run: |
make train_gpt2_fp32 test_gpt2_fp32
- name: Train model fp32
run: |
./train_gpt2_fp32
- name: Test model fp32
run: |
./test_gpt2_fp32gi

0 comments on commit 4e1c43d

Please sign in to comment.