github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-08 19:44:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(compiler): clarify input encryption noise variance

Browse Source
This commit is contained in:
Mayeul@Zama
2022-09-26 14:54:49 +02:00
committed by mayeul-zama
parent cb2c9ef6bf
commit f1833f06f2
3 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
compiler/include/concretelang/Conversion/Utils/GlobalFHEContext.h
View File

@@ -66,7 +66,7 @@ struct V0Parameter {
// TODO remove the shift when we have true polynomial size
size_t getPolynomialSize() { return 1 << logPolynomialSize; }
size_t getNBigGlweDimension() { return glweDimension * getPolynomialSize(); }
size_t getNBigLweDimension() { return glweDimension * getPolynomialSize(); }
};
struct V0FHEContext {

2
compiler/lib/Conversion/TFHEGlobalParametrization/TFHEGlobalParametrization.cpp
View File

@@ -67,7 +67,7 @@ public:
TFHE::GLWECipherTextType glweInterPBSType(GLWECipherTextType &type) {
auto bits = 64;
auto dimension = cryptoParameters.getNBigGlweDimension();
auto dimension = cryptoParameters.getNBigLweDimension();
auto polynomialSize = 1;
auto precision = (signed)type.getP();
auto crtDecomposition =

16
compiler/lib/Support/V0ClientParameters.cpp
View File

@@ -98,14 +98,16 @@ createClientParametersForV0(V0FHEContext fheContext,
llvm::StringRef functionName,
mlir::ModuleOp module) {
auto v0Param = fheContext.parameter;
Variance encryptionVariance = v0Curve->getVariance(
Variance inputVariance =
v0Curve->getVariance(1, v0Param.getNBigLweDimension(), 64);
Variance bootstrapKeyVariance = v0Curve->getVariance(
v0Param.glweDimension, v0Param.getPolynomialSize(), 64);
// Variance encryptionVariance = 0.;
Variance keyswitchVariance = v0Curve->getVariance(1, v0Param.nSmall, 64);
Variance keyswitchKeyVariance = v0Curve->getVariance(1, v0Param.nSmall, 64);
// Static client parameters from global parameters for v0
ClientParameters c;
c.secretKeys = {
{clientlib::BIG_KEY, {/*.size = */ v0Param.getNBigGlweDimension()}},
{clientlib::BIG_KEY, {/*.size = */ v0Param.getNBigLweDimension()}},
};
bool has_small_key = v0Param.nSmall != 0;
bool has_bootstrap = v0Param.brLevel != 0;
@@ -123,7 +125,7 @@ createClientParametersForV0(V0FHEContext fheContext,
/*.level = */ v0Param.brLevel,
/*.baseLog = */ v0Param.brLogBase,
/*.glweDimension = */ v0Param.glweDimension,
/*.variance = */ encryptionVariance,
/*.variance = */ bootstrapKeyVariance,
},
},
};
@@ -137,7 +139,7 @@ createClientParametersForV0(V0FHEContext fheContext,
/*.outputSecretKeyID = */ clientlib::SMALL_KEY,
/*.level = */ v0Param.ksLevel,
/*.baseLog = */ v0Param.ksLogBase,
/*.variance = */ keyswitchVariance,
/*.variance = */ keyswitchKeyVariance,
},
},
};
@@ -165,7 +167,7 @@ createClientParametersForV0(V0FHEContext fheContext,
: inputs.back().isa<mlir::concretelang::Concrete::ContextType>();
auto gateFromType = [&](mlir::Type ty) {
return gateFromMLIRType(clientlib::BIG_KEY, encryptionVariance, ty);
return gateFromMLIRType(clientlib::BIG_KEY, inputVariance, ty);
};
for (auto inType = funcType.getInputs().begin();
inType < funcType.getInputs().end() - hasContext; inType++) {