Guard two more calls to inspect

phanrahan · Apr 29, 2019 · 7641e86 · 7641e86
1 parent 75e4503
commit 7641e86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/magma/syntax/combinational.py b/magma/syntax/combinational.py
@@ -12,6 +12,7 @@
 import types
 from magma.debug import debug_info
 from magma.ssa import convert_tree_to_ssa
+from magma.config import get_debug_mode
 
 
 class CircuitDefinitionSyntaxError(Exception):
@@ -213,7 +214,7 @@ def combinational(defn_env: dict, fn: types.FunctionType):
     debug(source)
     circuit_def = ast_utils.compile_function_to_file(tree, fn.__name__,
                                                      defn_env)
-    if getattr(circuit_def, "debug_info", False):
+    if get_debug_mode() and getattr(circuit_def, "debug_info", False):
         circuit_def.debug_info = debug_info(circuit_def.debug_info.filename,
                                             circuit_def.debug_info.lineno,
                                             inspect.getmodule(fn))

diff --git a/magma/syntax/sequential.py b/magma/syntax/sequential.py
@@ -7,6 +7,7 @@
 import functools
 import magma as m
 from magma.ssa import convert_tree_to_ssa
+from magma.config import get_debug_mode
 from collections import Counter
 
 
@@ -355,7 +356,7 @@ def _sequential(defn_env: dict, cls):
             ast.parse("from mantle import DefineRegister").body[0],
         ] + tree.body)
     circuit_def = ast_utils.compile_function_to_file(tree, cls.__name__, defn_env)
-    if getattr(circuit_def, "debug_info", False):
+    if get_debug_mode() and getattr(circuit_def, "debug_info", False):
         circuit_def.debug_info = debug_info(circuit_def.debug_info.filename,
                                             circuit_def.debug_info.lineno,
                                             inspect.getmodule(cls))