Commit 9cec995
Tag PjRt migration candidates explicitly. (BUILD changes)
This change adds a new tag "pjrt_migration_candidate" to all test targets that
depend on HloTestBase, ClientLibraryTestBase, and HloRunnerTpuSystem.
This change also adds a new `use_legacy_runtime` kwarg to `xla_test`, which acts
as a replacement for "test_migrated_to_hlo_runner_pjrt". During a brief
transition phase, we will leave all "test_migrated_to_hlo_runner_pjrt" tags in
place so that we can identify any tests that have the `use_legacy_runtime` set
to an incorrect value.
"pjrt_migration_candidate" and "test_migrated_to_hlo_runner_pjrt" are mutually
exclusive.
"pjrt_migration_candidate" should not appear on any tests using the new runtime.
Unlike "test_migrated_to_hlo_runner_pjrt", which primarily tags `xla_test`
targets, "pjrt_migration_candidate" intends to tag all outstanding migration
candidates to obtain an accurate picture of migration progress. If a test cannot
or should not be migrated, it can be excluded from any analysis just by removing
the tag.
PiperOrigin-RevId: 8533054891 parent f0f6888 commit 9cec995
1 file changed
+2
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
152 | 152 | | |
153 | 153 | | |
154 | 154 | | |
| 155 | + | |
155 | 156 | | |
| 157 | + | |
156 | 158 | | |
157 | 159 | | |
158 | 160 | | |
| |||
0 commit comments