mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-11 05:04:58 -05:00
7a2511b3d4
* feat(compiler): low level fhe dialect * feat(compiler): using generated printer/parser in LowLFHE * feat(compiler): new types and ops for LowLFHE * tests(compiler): LowLFHE types and ops * feat(compiler): fill ops * cleanup * summary + description * tests(compiler): use new CLI args * formatting
67 lines
1.8 KiB
C++
67 lines
1.8 KiB
C++
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEDialect.h"
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEOps.h"
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHETypes.h"
|
|
|
|
#define GET_TYPEDEF_CLASSES
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEOpsTypes.cpp.inc"
|
|
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEOpsDialect.cpp.inc"
|
|
|
|
using namespace mlir::zamalang::LowLFHE;
|
|
|
|
void LowLFHEDialect::initialize() {
|
|
addOperations<
|
|
#define GET_OP_LIST
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEOps.cpp.inc"
|
|
>();
|
|
|
|
addTypes<
|
|
#define GET_TYPEDEF_LIST
|
|
#include "zamalang/Dialect/LowLFHE/IR/LowLFHEOpsTypes.cpp.inc"
|
|
>();
|
|
}
|
|
|
|
::mlir::Type LowLFHEDialect::parseType(::mlir::DialectAsmParser &parser) const {
|
|
mlir::Type type;
|
|
|
|
std::string types_str[] = {
|
|
"enc_rand_gen",
|
|
"secret_rand_gen",
|
|
"plaintext",
|
|
"plaintext_list",
|
|
"foreign_plaintext_list",
|
|
"lwe_ciphertext",
|
|
"lwe_key_switch_key",
|
|
"lwe_bootstrap_key",
|
|
"lwe_secret_key",
|
|
"lwe_size",
|
|
"glwe_ciphertext",
|
|
"glwe_secret_key",
|
|
"glwe_size",
|
|
"polynomial_size",
|
|
"decomp_level_count",
|
|
"decomp_base_log",
|
|
"variance",
|
|
"cleartext",
|
|
};
|
|
|
|
for (const std::string &type_str : types_str) {
|
|
if (parser.parseOptionalKeyword(type_str).succeeded()) {
|
|
generatedTypeParser(this->getContext(), parser, type_str, type);
|
|
return type;
|
|
}
|
|
}
|
|
|
|
parser.emitError(parser.getCurrentLocation(), "Unknown LowLFHE type");
|
|
// call default parser
|
|
parser.parseType(type);
|
|
return type;
|
|
}
|
|
|
|
void LowLFHEDialect::printType(::mlir::Type type,
|
|
::mlir::DialectAsmPrinter &printer) const {
|
|
if (generatedTypePrinter(type, printer).failed())
|
|
// Calling default printer if failed to print LowLFHE type
|
|
printer.printType(type);
|
|
}
|