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

feat(compiler): Add verifier to TFHE.encode_expand_lut_for_bootstrap

Browse Source
This commit is contained in:
Andi Drebes
2024-01-29 11:55:25 +01:00
parent ea07239732
commit 8d71dc2028
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
compilers/concrete-compiler/compiler/include/concretelang/Dialect/TFHE/IR/TFHEOps.td
View File

@@ -100,6 +100,8 @@ def TFHE_EncodeExpandLutForBootstrapOp : TFHE_Op<"encode_expand_lut_for_bootstra
);
let results = (outs 1DTensorOf<[I64]> : $result);
let hasVerifier = 1;
}
def TFHE_EncodeLutForCrtWopPBSOp : TFHE_Op<"encode_lut_for_crt_woppbs", [Pure]> {

18
compilers/concrete-compiler/compiler/lib/Dialect/TFHE/IR/TFHEOps.cpp
View File

@@ -117,6 +117,24 @@ mlir::LogicalResult MulGLWEIntOp::verify() {
*this);
}
mlir::LogicalResult EncodeExpandLutForBootstrapOp::verify() {
mlir::IntegerAttr polySizeAttr = this->getPolySizeAttr();
mlir::RankedTensorType rtt =
this->getResult().getType().template cast<mlir::RankedTensorType>();
if (rtt.getNumElements() != polySizeAttr.getInt()) {
this->emitError("The number of elements of the output tensor of ")
<< rtt.getNumElements()
<< " does not match the size of the polynomial of "
<< polySizeAttr.getInt();
return mlir::failure();
}
return mlir::success();
}
template <typename BootstrapOpT>
mlir::LogicalResult verifyBootstrapSingleLUTConstraints(BootstrapOpT &op) {
GLWEBootstrapKeyAttr keyAttr = op.getKeyAttr();