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(compiler): Add !MidLFHE.ggsw<size,N,level,base_log>

Browse Source
This commit is contained in:
Quentin Bourgerie
2021-05-25 10:46:34 +02:00
parent 122759eabf
commit ce69aab756
3 changed files with 60 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
compiler/include/zamalang/Dialect/MidLFHE/IR/MidLFHETypes.td
View File

@@ -68,4 +68,48 @@ def GLWECipherTextType : MidLFHE_Type<"GLWECipherText"> {
}];
}
def GGSWCipherTextType : MidLFHE_Type<"GGSWCipherText"> {
let mnemonic = "ggsw";
let summary = "An GGSW cipher text";
let description = [{
An GGSW cipher text
}];
let parameters = (ins "unsigned":$size, "unsigned":$N, "unsigned": $level, "unsigned":$base_log);
// We define the printer inline.
let printer = [{
$_printer << "ggsw<" << getImpl()->size << "," << getImpl()->N << "," << getImpl()->level << "," << getImpl()->base_log << ">";
}];
// The parser is defined here also.
let parser = [{
if ($_parser.parseLess())
return Type();
int size;
if ($_parser.parseInteger(size))
return Type();
if ($_parser.parseComma())
return Type();
int N;
if ($_parser.parseInteger(N))
return Type();
if ($_parser.parseComma())
return Type();
int level;
if ($_parser.parseInteger(level))
return Type();
if ($_parser.parseComma())
return Type();
int base_log;
if ($_parser.parseInteger(base_log))
return Type();
if ($_parser.parseGreater())
return Type();
return get($_ctxt, size, N, level, base_log);
}];
}
#endif

12
compiler/lib/Dialect/MidLFHE/IR/MidLFHEDialect.cpp
View File

@@ -21,11 +21,12 @@ void MidLFHEDialect::initialize() {
::mlir::Type MidLFHEDialect::parseType(::mlir::DialectAsmParser &parser) const
{
if(parser.parseOptionalKeyword("glwe").succeeded())
return GLWECipherTextType::parse(this->getContext(), parser);
if(parser.parseOptionalKeyword("lwe").succeeded())
return LWECipherTextType::parse(this->getContext(), parser);
if(parser.parseOptionalKeyword("glwe").succeeded())
return GLWECipherTextType::parse(this->getContext(), parser);
if(parser.parseOptionalKeyword("ggsw").succeeded())
return GGSWCipherTextType::parse(this->getContext(), parser);
return ::mlir::Type();
}
@@ -42,6 +43,11 @@ void MidLFHEDialect::printType(::mlir::Type type,
glwe.print(printer);
return;
}
mlir::zamalang::MidLFHE::GGSWCipherTextType ggsw = type.dyn_cast_or_null<mlir::zamalang::MidLFHE::GGSWCipherTextType>();
if (ggsw != nullptr) {
ggsw.print(printer);
return;
}
// TODO - What should be done here?
printer << "unknwontype";
}

7
compiler/tests/Dialect/MidLFHE/types_ggsw.mlir Normal file
View File

@@ -0,0 +1,7 @@
// RUN: zamacompiler %s 2>&1| FileCheck %s
// CHECK-LABEL: func @ggsw(%arg0: !MidLFHE.ggsw<1024,12,3,2>) -> !MidLFHE.ggsw<1024,12,3,2>
func @ggsw(%arg0: !MidLFHE.ggsw<1024,12,3,2>) -> !MidLFHE.ggsw<1024,12,3,2> {
// CHECK-LABEL: return %arg0 : !MidLFHE.ggsw<1024,12,3,2>
return %arg0: !MidLFHE.ggsw<1024,12,3,2>
}