Skip to content

Commit 9cec995

Browse files
nvgrwGoogle-ML-Automation
authored andcommitted
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: 853305489
1 parent f0f6888 commit 9cec995

File tree

1 file changed

+2
-0
lines changed
  • xla/stream_executor/sycl

1 file changed

+2
-0
lines changed

xla/stream_executor/sycl/BUILD

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,9 @@ xla_test(
152152
tags = [
153153
"gpu",
154154
"oneapi-only",
155+
"pjrt_migration_candidate",
155156
],
157+
use_legacy_runtime = True,
156158
deps = [
157159
":sycl_event",
158160
":sycl_platform_id",

0 commit comments

Comments
 (0)