Skip to content

Commit c9ebf4b

Browse files
authored
enable v1 tpu test (#72)
Signed-off-by: Richard Liu <ricliu@google.com>
1 parent eff0145 commit c9ebf4b

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

scripts/test-template-fastcheck.j2

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ steps:
219219
{% endif %}
220220
{% endfor %}
221221

222-
- label: "TPU Test"
222+
- label: "TPU V0 Test"
223223
depends_on: ~
224224
soft_fail: true
225225
agents:
@@ -228,6 +228,15 @@ steps:
228228
- if [[ -f ".buildkite/run-tpu-test.sh" ]]; then bash .buildkite/run-tpu-test.sh; fi
229229
- yes | docker system prune -a
230230

231+
- label: "TPU V1 Test"
232+
depends_on: ~
233+
soft_fail: true
234+
agents:
235+
queue: tpu_queue
236+
commands:
237+
- if [[ -f ".buildkite/run-tpu-v1-test.sh" ]]; then bash .buildkite/run-tpu-v1-test.sh; fi
238+
- yes | docker system prune -a
239+
231240
- block: "Run GH200 Test"
232241
depends_on: ~
233242
key: block-gh200

0 commit comments

Comments
 (0)