-
Notifications
You must be signed in to change notification settings - Fork 0
44 lines (36 loc) · 1.02 KB
/
ci-train-gpu.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
name: GPU Train and Test
on: [push, pull_request]
jobs:
diff:
runs-on: ubuntu-latest
outputs:
train_test_changed: ${{ steps.filter.outputs.train_test_changed }}
steps:
- uses: actions/checkout@v4
- uses: dorny/paths-filter@v3
id: filter
with:
filters: |
train_test_changed:
- 'train_*'
- 'test_*'
train-and-test-gpu:
needs: diff
runs-on: devcloud # self-hosted GPU runner on Intel devcloud
if: needs.diff.outputs.train_test_changed == 'true'
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_fp32