From e223c0a00ef25fa658f7766c08ba07ed2ec5841b Mon Sep 17 00:00:00 2001 From: bamsumit Date: Wed, 21 Feb 2024 17:49:16 -0800 Subject: [PATCH] caching fixes Signed-off-by: bamsumit --- src/lava/magma/compiler/compiler.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/lava/magma/compiler/compiler.py b/src/lava/magma/compiler/compiler.py index 0c0d21ff6..fd41bd221 100644 --- a/src/lava/magma/compiler/compiler.py +++ b/src/lava/magma/compiler/compiler.py @@ -239,8 +239,11 @@ def _compile_proc_groups( for proc_name, pb in proc_builders_values.items(): proc = procname_to_proc_map[proc_name] proc_builders[proc] = pb + pb.proc_params = proc.proc_params channel_map.read_from_cache(cache_object, procname_to_proc_map) + print(f"\nBuilders and Channel Map loaded from " \ + f"Cache {cache_dir}\n") return proc_builders, channel_map # Create the global ChannelMap that is passed between @@ -286,10 +289,14 @@ def _compile_proc_groups( f"Violation Name: {p.name=}" raise Exception(msg) procname_to_proc_builder[p.name] = pb + pb.proc_params = None cache_object["procname_to_proc_builder"] = procname_to_proc_builder channel_map.write_to_cache(cache_object, proc_to_procname_map) with open(os.path.join(cache_dir, "cache"), "wb") as cache_file: pickle.dump(cache_object, cache_file) + for p, pb in proc_builders.items(): + pb.proc_params = p.proc_params + print(f"\nBuilders and Channel Map stored to Cache {cache_dir}\n") return proc_builders, channel_map @staticmethod