variables: fix stage lookup gaffe

Signed-off-by: Øyvind Harboe <[email protected]>
The-OpenROAD-Project · Oct 12, 2024 · f0aca8f · f0aca8f
1 parent bcf2ded
commit f0aca8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openroad.bzl b/openroad.bzl
@@ -1214,7 +1214,7 @@ def get_stage_args(stage, stage_arguments, arguments):
     unsorted_dict = ({
                          arg: value
                          for arg, value in arguments.items()
-                         if arg in ALL_STAGE_TO_VARIABLES[stage]
+                         if arg in ALL_STAGE_TO_VARIABLES[stage] or arg not in ALL_VARIABLES_TO_STAGE
                      } |
                      stage_arguments.get(stage, {}))
     return dict(sorted(unsorted_dict.items()))
@@ -1315,7 +1315,7 @@ def orfs_flow(
 
     mock_stage_arguments = _merge(
         {stage: {arg: value for arg, value in arguments.items() if arg in stage_args} for stage, stage_args in ALL_STAGE_TO_VARIABLES.items()},
-        {stage: {arg: value for arg, value in arguments.items() if stage in ALL_VARIABLES_TO_STAGE.get(arg, [])} for stage in ALL_STAGES},
+        {stage: {arg: value for arg, value in arguments.items() if stage in ALL_VARIABLES_TO_STAGE.get(arg, ALL_STAGES)} for stage in ALL_STAGES},
         stage_arguments,
         MOCK_STAGE_ARGUMENTS,
     )