setup xl-ml test for 2.2 release (#1019)

* setup xl-ml test for 2.2 release * remove comment * remove xrt * typo fix * typo fix * update tag * update tag * fix pt2.2 tests based on the nightly test * update image for gpu * update pytorch dependency and app tag name * fix libtpu.so issue
GoogleCloudPlatform · Dec 12, 2023 · 669b136 · 669b136
1 parent 3a3911f
commit 669b136
Show file tree

Hide file tree

Showing 15 changed files with 163 additions and 630 deletions.
diff --git a/dashboard/app.yaml b/dashboard/app.yaml
@@ -33,7 +33,7 @@ resources:
 env_variables:
   REDISHOST: '10.25.27.107'
   REDISPORT: '6379'
-  TEST_NAME_PREFIXES: 'pt-nightly,pt-2.1,tf.nightly,tf.se.nightly,%-1vm,jax,flax,pax-nightly,mp-jax,mp-pax,mp-pt'
+  TEST_NAME_PREFIXES: 'pt-nightly,pt-2.2,tf.nightly,tf.se.nightly,%-1vm,jax,flax,pax-nightly,mp-jax,mp-pax,mp-pt'
   JOB_HISTORY_TABLE_NAME: 'xl-ml-test.metrics_handler_dataset.job_history'
   METRIC_HISTORY_TABLE_NAME: 'xl-ml-test.metrics_handler_dataset.metric_history'
 

diff --git a/tests/pytorch/r2.1/hf-mae.libsonnet b/tests/pytorch/r2.1/hf-mae.libsonnet