|
8 | 8 | types: [opened, reopened, ready_for_review, synchronize] # added `ready_for_review` since draft is skipped
|
9 | 9 | paths:
|
10 | 10 | - ".github/workflows/tpu-tests.yml"
|
| 11 | + - "dockers/base-xla/*" |
11 | 12 | - "requirements/lite/**"
|
12 | 13 | - "src/lightning_lite/**"
|
13 | 14 | - "tests/tests_lite/**"
|
|
33 | 34 | if: github.event.pull_request.draft == false
|
34 | 35 | env:
|
35 | 36 | PYTHON_VER: 3.7
|
36 |
| - timeout-minutes: 100 # should match the timeout in `tpu_test_cases.jsonnet` |
| 37 | + timeout-minutes: 100 # should match the timeout in `tpu_workflow.jsonnet` |
37 | 38 |
|
38 | 39 | steps:
|
39 | 40 | - uses: actions/checkout@v3
|
|
64 | 65 | PR_NUMBER: ${{ github.event.pull_request.number }}
|
65 | 66 | SHA: ${{ github.event.pull_request.head.sha }}
|
66 | 67 | run: |
|
67 |
| - python -c "fname = 'dockers/tpu-tests/tpu_test_cases.jsonnet' ; data = open(fname).read().replace('{PYTORCH_VERSION}', '$XLA_VER') |
| 68 | + python -c "fname = 'dockers/base-xla/tpu_workflow.jsonnet' ; data = open(fname).read().replace('{PYTORCH_VERSION}', '$XLA_VER') |
68 | 69 | data = data.replace('{PYTHON_VERSION}', '$PYTHON_VER').replace('{PR_NUMBER}', '$PR_NUMBER').replace('{SHA}', '$SHA') ; open(fname, 'w').write(data)"
|
69 |
| - cat dockers/tpu-tests/tpu_test_cases.jsonnet |
| 70 | + cat dockers/base-xla/tpu_workflow.jsonnet |
70 | 71 | shell: bash
|
71 | 72 |
|
72 | 73 | - uses: google-github-actions/auth@v0
|
|
82 | 83 | - name: Deploy cluster
|
83 | 84 | run: |
|
84 | 85 | export PATH=$PATH:$HOME/go/bin
|
85 |
| - job_name=$(jsonnet -J ml-testing-accelerators/ dockers/tpu-tests/tpu_test_cases.jsonnet | kubectl create -f -) |
| 86 | + job_name=$(jsonnet -J ml-testing-accelerators/ dockers/base-xla/tpu_workflow.jsonnet | kubectl create -f -) |
86 | 87 | job_name=${job_name#job.batch/}
|
87 | 88 | job_name=${job_name% created}
|
88 | 89 | pod_name=$(kubectl get po -l controller-uid=`kubectl get job $job_name -o "jsonpath={.metadata.labels.controller-uid}"` | awk 'match($0,!/NAME/) {print $1}')
|
|
0 commit comments