Set environment variables for cocotb 2.0 compatibility

olofk · Nov 26, 2024 · 4502827 · 4502827
1 parent 602f134
commit 4502827
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/edalize/flows/sim.py b/edalize/flows/sim.py
@@ -60,5 +60,9 @@ def run(self, args=None):
         # Get run command from simulator
         (cmd, args, cwd) = run_tool.run()
         cocotb_module = self.flow_options.get("cocotb_module")
-        env = {"MODULE": cocotb_module} if cocotb_module else {}
+        env = (
+            {"MODULE": cocotb_module, "COCOTB_TEST_MODULES": cocotb_module}
+            if cocotb_module
+            else {}
+        )
         self._run_tool(cmd, args=args, cwd=cwd, env=env)