caching coreir context with debugging code

magma/frontend/coreir_.py

@@ -1,7 +1,17 @@
-from magma.backend.coreir_ import CoreIRBackend
+from magma import cache_definition
+from magma.backend.coreir_ import CoreIRBackend, magma_coreir_context
 from magma.circuit import DefineCircuitKind, Circuit
 from magma import cache_definition
 from coreir.generator import Generator
+import coreir
+
+@cache_definition
+def GetCoreIRBackend():
+    return CoreIRBackend()
+
+@cache_definition
+def GetMagmaContext():
+    return magma_coreir_context
 
 def DefineModuleWrapper(cirb: CoreIRBackend, coreirModule, uniqueName, deps):
     class ModuleWrapper(Circuit):
@@ -18,8 +28,15 @@ def definition(cls):
 
 def DefineCircuitFromGeneratorWrapper(cirb: CoreIRBackend, namespace: str, generator: str,
                                       uniqueName: str, dependentNamespaces: list = [],
-                                      genargs: dict = {}, runGenerators = True):
+                                      genargs: dict = {}, runGenerators = True, print_error=False):
     moduleToWrap = cirb.context.import_generator(namespace,generator)(**genargs)
+
+    if print_error:
+        print(f"Cached cirb IO: {moduleToWrap.type.items()}")
+        fresh_context = coreir.Context()
+        fresh_moduleToWrap = fresh_context.import_generator(namespace,generator)(**genargs)
+        print(f"Fresh cirb IO: {fresh_moduleToWrap.type.items()}")
+
     deps = [namespace] + dependentNamespaces
     if runGenerators:
         cirb.context.run_passes(["rungenerators"], deps)

magma/simulator/coreir_simulator.py

@@ -3,6 +3,7 @@
 
 from .simulator import CircuitSimulator, ExecutionState
 from ..backend import coreir_
+from ..frontend.coreir_ import GetMagmaContext
 from ..scope import Scope
 from ..ref import DefnRef, ArrayRef, TupleRef
 from ..array import ArrayType
@@ -134,7 +135,7 @@ def __init__(self, circuit, clock, coreir_filename=None, context=None,
         setup_clocks(circuit)
 
         if context is None:
-            self.ctx = coreir.Context()
+            self.ctx = GetMagmaContext()
         else:
             self.ctx = context
         coreir_.compile(circuit, coreir_filename, context=self.ctx)