|
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