Handle empty string when using getenv (pytorch#137159)

andrasmatyassy · Oct 31, 2024 · e34366d · e34366d
1 parent c3bff46
commit e34366d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/torch/fx/experimental/symbolic_shapes.py b/torch/fx/experimental/symbolic_shapes.py
@@ -4281,15 +4281,15 @@ def create_symbol(
                 sympy_expr
             ) in config.extended_debug_create_symbol.split(",")
             maybe_more_info = ""
-            if (
-                not is_debug
-                and os.getenv("TORCHDYNAMO_EXTENDED_ADVICE=0") is not None
+            if not is_debug and os.getenv("TORCHDYNAMO_EXTENDED_ADVICE", "1") not in (
+                "0",
+                "",
             ):
                 maybe_more_info = (
                     ", for more info run with "
                     f'TORCHDYNAMO_EXTENDED_DEBUG_CREATE_SYMBOL="{sympy_expr}" '
-                    "or suppress this message by setting the environment "
-                    'variable "TORCHDYNAMO_EXTENDED_ADVICE=0"'
+                    "or to suppress this message run with "
+                    'TORCHDYNAMO_EXTENDED_ADVICE="0"'
                 )
             sloc, maybe_extra_debug = self._get_stack_summary(is_debug)
             self.log.info(