fix flag

pytorch · May 10, 2024 · 37facf1 · 37facf1
1 parent 8fff517
commit 37facf1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torch_xla/stablehlo.py b/torch_xla/stablehlo.py
@@ -347,13 +347,13 @@ def _exported_program_to_stablehlo_bundle(exported_model,
   if options.inline_all_constant:
     # Inline all constants.
     torch_xla._XLAC._set_xla_all_numbers_special_scalars(True)
+  if "XLA_HLO_DEBUG" in os.environ:
+    xla_hlo_debug_env = os.environ["XLA_HLO_DEBUG"]
+  else:
+    xla_hlo_debug_env = "0"
   if options.export_node_metadata:
     gm_serializer = GraphModuleSerializer(exported_model.graph_signature,
                                           exported_model.module_call_graph)
-    if "XLA_HLO_DEBUG" in os.environ:
-      xla_hlo_debug_env = os.environ["XLA_HLO_DEBUG"]
-    else:
-      xla_hlo_debug_env = "0"
     os.environ["XLA_HLO_DEBUG"] = "1"
   else:
     gm_serializer = None