mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
feat(compiler): Determine FHE circuit constraints instead of using default values
This replaces the default FHE circuit constrains (maximum encrypted integer width of 7 bits and a Minimal Arithmetic Noise Padding of 10 with the results of the `MaxMANP` pass, which determines these values automatically from the input program. Since the maximum encrypted integer width and the maximum value for the Minimal Arithmetic Noise Padding can only be derived from HLFHE operations, the circuit constraints are determined automatically by `zamacompiler` only if the option `--entry-dialect=hlfhe` was specified. For lower-level dialects, `zamacompiler` has been provided with the options `--assume-max-eint-precision=...` and `--assume-max-manp=...` that allow a user to specify the values for the maximum required precision and maximum values for the Minimal Arithmetic Noise Padding.
This commit is contained in:
committed by
Quentin Bourgerie
parent
2ed1720234
commit
2acfa63eb7
@@ -13,6 +13,7 @@
|
||||
#include <zamalang/Dialect/HLFHE/Analysis/MANP.h>
|
||||
#include <zamalang/Support/Pipeline.h>
|
||||
#include <zamalang/Support/logging.h>
|
||||
#include <zamalang/Support/math.h>
|
||||
|
||||
namespace mlir {
|
||||
namespace zamalang {
|
||||
@@ -35,6 +36,51 @@ mlir::LogicalResult invokeMANPPass(mlir::MLIRContext &context,
|
||||
return pm.run(module);
|
||||
}
|
||||
|
||||
llvm::Expected<llvm::Optional<mlir::zamalang::V0FHEConstraint>>
|
||||
getFHEConstraintsFromHLFHE(mlir::MLIRContext &context, mlir::ModuleOp &module) {
|
||||
llvm::Optional<size_t> oMax2norm;
|
||||
llvm::Optional<size_t> oMaxWidth;
|
||||
|
||||
mlir::PassManager pm(&context);
|
||||
|
||||
addPotentiallyNestedPass(pm, mlir::zamalang::createMANPPass());
|
||||
addPotentiallyNestedPass(
|
||||
pm, mlir::zamalang::createMaxMANPPass([&](const llvm::APInt &currMaxMANP,
|
||||
unsigned currMaxWidth) {
|
||||
assert((uint64_t)currMaxWidth < std::numeric_limits<size_t>::max() &&
|
||||
"Maximum width does not fit into size_t");
|
||||
|
||||
assert(sizeof(uint64_t) >= sizeof(size_t) &&
|
||||
currMaxMANP.ult(std::numeric_limits<size_t>::max()) &&
|
||||
"Maximum MANP does not fit into size_t");
|
||||
|
||||
size_t manp = (size_t)currMaxMANP.getZExtValue();
|
||||
size_t width = (size_t)currMaxWidth;
|
||||
|
||||
if (!oMax2norm.hasValue() || oMax2norm.getValue() < manp)
|
||||
oMax2norm.emplace(manp);
|
||||
|
||||
if (!oMaxWidth.hasValue() || oMaxWidth.getValue() < width)
|
||||
oMaxWidth.emplace(width);
|
||||
}));
|
||||
|
||||
if (pm.run(module.getOperation()).failed()) {
|
||||
return llvm::make_error<llvm::StringError>(
|
||||
"Failed to determine the maximum Arithmetic Noise Padding and maximum"
|
||||
"required precision",
|
||||
llvm::inconvertibleErrorCode());
|
||||
}
|
||||
|
||||
llvm::Optional<mlir::zamalang::V0FHEConstraint> ret;
|
||||
|
||||
if (oMax2norm.hasValue() && oMaxWidth.hasValue()) {
|
||||
ret = llvm::Optional<mlir::zamalang::V0FHEConstraint>(
|
||||
{.norm2 = ceilLog2(oMax2norm.getValue()), .p = oMaxWidth.getValue()});
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
mlir::LogicalResult lowerHLFHEToMidLFHE(mlir::MLIRContext &context,
|
||||
mlir::ModuleOp &module, bool verbose) {
|
||||
mlir::PassManager pm(&context);
|
||||
|
||||
Reference in New Issue
Block a user