github/Sunscreen
1
1
Fork 0
mirror of https://github.com/Sunscreen-tech/Sunscreen.git synced 2026-04-19 03:00:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix mutability issue in fhe program macro

Browse Source
This commit is contained in:
Matthew Liu
2023-08-01 16:03:29 -07:00
parent 130fa517bf
commit a137c5109b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sunscreen_compiler_macros/src/fhe_program.rs
View File

@@ -141,7 +141,7 @@ pub fn fhe_program_impl(
}
// TODO: Other schemes.
let mut context = FheContext::new(params.clone());
let mut context = ContextEnum::Fhe(FheContext::new(params.clone()));
CURRENT_PROGRAM_CTX.with(|ctx| {
#[allow(clippy::type_complexity)]
@@ -153,7 +153,7 @@ pub fn fhe_program_impl(
// Transmute away the lifetime to 'static. So long as we are careful with internal()
// panicing, this is safe because we set the context back to none before the funtion
// returns.
ctx.swap(&RefCell::new(Some(unsafe { transmute(&mut ContextEnum::Fhe(context.clone())) })));
ctx.swap(&RefCell::new(Some(unsafe { transmute(&mut context) })));
#(#var_decl)*
@@ -180,7 +180,7 @@ pub fn fhe_program_impl(
ctx.swap(&RefCell::new(None));
});
Ok(context.clone().graph)
Ok(context.unwrap_fhe().unwrap().graph.clone())
}
fn signature(&self) -> sunscreen::CallSignature {