github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement MANP calculation for tensor.insert op

Browse Source
This commit is contained in:
Umut
2022-06-21 09:40:51 +02:00
parent bcc5c750a1
commit 8f8a57d220
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
compiler/lib/Dialect/FHE/Analysis/MANP.cpp
View File

@@ -966,6 +966,20 @@ static llvm::APInt getSqMANP(
operandMANPs[1]->getValue().getMANP().getValue());
}
static llvm::APInt getSqMANP(
mlir::tensor::InsertOp op,
llvm::ArrayRef<mlir::LatticeElement<MANPLatticeValue> *> operandMANPs) {
assert(
operandMANPs.size() >= 2 &&
operandMANPs[0]->getValue().getMANP().hasValue() &&
operandMANPs[1]->getValue().getMANP().hasValue() &&
"Missing squared Minimal Arithmetic Noise Padding for encrypted operand");
return APIntUMax(operandMANPs[0]->getValue().getMANP().getValue(),
operandMANPs[1]->getValue().getMANP().getValue());
}
static llvm::APInt getSqMANP(
mlir::tensor::CollapseShapeOp op,
llvm::ArrayRef<mlir::LatticeElement<MANPLatticeValue> *> operandMANPs) {
@@ -1274,6 +1288,18 @@ struct MANPAnalysis : public mlir::ForwardDataFlowAnalysis<MANPLatticeValue> {
isDummy = true;
}
}
// InsertOp
else if (auto insertOp = llvm::dyn_cast<mlir::tensor::InsertOp>(op)) {
if (insertOp.result()
.getType()
.cast<mlir::TensorType>()
.getElementType()
.isa<mlir::concretelang::FHE::EncryptedIntegerType>()) {
norm2SqEquiv = getSqMANP(insertOp, operands);
} else {
isDummy = true;
}
}
// InsertSliceOp
else if (auto insertSliceOp =
llvm::dyn_cast<mlir::tensor::InsertSliceOp>(op)) {