Set a global fork graph in program cache

core/src/replay_stage.rs

@@ -3859,6 +3859,7 @@ impl ReplayStage {
         epoch_slots_frozen_slots: &mut EpochSlotsFrozenSlots,
         drop_bank_sender: &Sender<Vec<Arc<Bank>>>,
     ) {
+        bank_forks.read().unwrap().prune_program_cache(new_root);
         let removed_banks = bank_forks.write().unwrap().set_root(
             new_root,
             accounts_background_request_sender,

ledger/src/bank_forks_utils.rs

@@ -189,6 +189,15 @@ pub fn load_bank_forks(
             (bank_forks, None)
         };
 
+    bank_forks
+        .read()
+        .expect("Failed to read lock the bank forks")
+        .root_bank()
+        .loaded_programs_cache
+        .write()
+        .expect("Failed to write lock the program cache")
+        .set_fork_graph(bank_forks.clone());
+
     let mut leader_schedule_cache =
         LeaderScheduleCache::new_from_bank(&bank_forks.read().unwrap().root_bank());
     if process_options.full_leader_cache {

ledger/src/blockstore_processor.rs

@@ -1555,6 +1555,11 @@ fn load_frozen_forks(
                 root = new_root_bank.slot();
 
                 leader_schedule_cache.set_root(new_root_bank);
+                new_root_bank
+                    .loaded_programs_cache
+                    .write()
+                    .unwrap()
+                    .prune(root, new_root_bank.epoch());
                 let _ = bank_forks.write().unwrap().set_root(
                     root,
                     accounts_background_request_sender,