aot: Ensure that all direct calls have compiled functions

dinfuehr · Jan 3, 2025 · 441fc27 · 441fc27
1 parent 9589e90
commit 441fc27
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/dora-runtime/src/compiler/aot.rs b/dora-runtime/src/compiler/aot.rs
@@ -520,14 +520,16 @@ fn prepare_lazy_call_sites(_vm: &VM, ctc: &CompiledTransitiveClosure) {
                     type_params,
                     const_pool_offset_from_ra,
                 } => {
-                    let address = ctc.function_addresses.get(&(*fct_id, type_params.clone()));
-                    if let Some(address) = address {
-                        let ra = code.instruction_start().offset(*offset as usize);
-                        let const_pool_address = ra.ioffset(*const_pool_offset_from_ra as isize);
+                    let address = ctc
+                        .function_addresses
+                        .get(&(*fct_id, type_params.clone()))
+                        .cloned()
+                        .expect("missing function");
+                    let ra = code.instruction_start().offset(*offset as usize);
+                    let const_pool_address = ra.ioffset(*const_pool_offset_from_ra as isize);
 
-                        unsafe {
-                            *const_pool_address.to_mut_ptr::<Address>() = *address;
-                        }
+                    unsafe {
+                        *const_pool_address.to_mut_ptr::<Address>() = address;
                     }
                 }