[XLA:GPU] Allow auto layout in multihost HLO runner.

PiperOrigin-RevId: 696486224
openxla · Nov 15, 2024 · d7682a0 · d7682a0
1 parent 10769ee
commit d7682a0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xla/tools/multihost_hlo_runner/functional_hlo_runner.cc b/xla/tools/multihost_hlo_runner/functional_hlo_runner.cc
@@ -332,6 +332,10 @@ absl::StatusOr<CompileOptions> FunctionalHloRunner::CreateCompileOptions(
     debug_options.set_xla_dump_hlo_as_proto(raw_options.xla_proto_dump_mode ==
                                             XlaProtoDumpMode::kDumpAsProto);
   }
+  // TODO(b/378832773) The flag is temporary, this line should be removed when
+  // default is changed / or flag is removed.
+  build_options.mutable_debug_options()->set_xla_pjrt_allow_auto_layout_in_hlo(
+      true);
 
   switch (raw_options.hlo_passes_mode) {
     case HloPassesMode::kRunXLABackendOnly: