diff --git a/compiler/lib/Runtime/context.c b/compiler/lib/Runtime/context.c index 8da6d7e4f..407b0da29 100644 --- a/compiler/lib/Runtime/context.c +++ b/compiler/lib/Runtime/context.c @@ -1,3 +1,4 @@ +#include "concrete-ffi.h" #include "zamalang/Runtime/context.h" #include @@ -6,5 +7,10 @@ LweKeyswitchKey_u64 *get_keyswitch_key(RuntimeContext *context) { } LweBootstrapKey_u64 *get_bootstrap_key(RuntimeContext *context) { - return context->bsk; + int err; + LweBootstrapKey_u64 *clone = + clone_lwe_bootstrap_key_u64(&err, context->bsk); + if (err != 0) + fprintf(stderr, "Runtime: cloning bootstrap key failed.\n"); + return clone; }