mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 12:15:09 -05:00
146 lines
3.6 KiB
C++
146 lines
3.6 KiB
C++
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
|
|
// Exceptions. See
|
|
// https://github.com/zama-ai/concrete-compiler-internal/blob/main/LICENSE.txt
|
|
// for license information.
|
|
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteDialect.h"
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteOps.h"
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteOpsDialect.cpp.inc"
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteTypes.h"
|
|
|
|
#define GET_TYPEDEF_CLASSES
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteOpsTypes.cpp.inc"
|
|
|
|
using namespace mlir::concretelang::Concrete;
|
|
|
|
void ConcreteDialect::initialize() {
|
|
addOperations<
|
|
#define GET_OP_LIST
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteOps.cpp.inc"
|
|
>();
|
|
|
|
addTypes<
|
|
#define GET_TYPEDEF_LIST
|
|
#include "concretelang/Dialect/Concrete/IR/ConcreteOpsTypes.cpp.inc"
|
|
>();
|
|
}
|
|
|
|
void printSigned(mlir::AsmPrinter &p, signed i) {
|
|
if (i == -1)
|
|
p << "_";
|
|
else
|
|
p << i;
|
|
}
|
|
|
|
mlir::Type GlweCiphertextType::parse(mlir::AsmParser &parser) {
|
|
if (parser.parseLess())
|
|
return Type();
|
|
int glweDimension = -1;
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(glweDimension))
|
|
return Type();
|
|
if (parser.parseComma())
|
|
return Type();
|
|
int polynomialSize = -1;
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(polynomialSize))
|
|
return Type();
|
|
if (parser.parseComma())
|
|
return Type();
|
|
|
|
int p = -1;
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(p))
|
|
return Type();
|
|
if (parser.parseGreater())
|
|
return Type();
|
|
Location loc = parser.getEncodedSourceLoc(parser.getNameLoc());
|
|
return getChecked(loc, loc.getContext(), glweDimension, polynomialSize, p);
|
|
}
|
|
|
|
void GlweCiphertextType::print(mlir::AsmPrinter &p) const {
|
|
p << "<";
|
|
printSigned(p, getGlweDimension());
|
|
p << ",";
|
|
printSigned(p, getPolynomialSize());
|
|
p << ",";
|
|
printSigned(p, getP());
|
|
p << ">";
|
|
}
|
|
|
|
void LweCiphertextType::print(mlir::AsmPrinter &p) const {
|
|
p << "<";
|
|
printSigned(p, getDimension());
|
|
p << ",";
|
|
printSigned(p, getP());
|
|
p << ">";
|
|
}
|
|
|
|
mlir::Type LweCiphertextType::parse(mlir::AsmParser &parser) {
|
|
if (parser.parseLess())
|
|
return mlir::Type();
|
|
|
|
int dimension = -1;
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(dimension))
|
|
return mlir::Type();
|
|
if (parser.parseComma())
|
|
return mlir::Type();
|
|
int p = -1;
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(p))
|
|
return mlir::Type();
|
|
if (parser.parseGreater())
|
|
return mlir::Type();
|
|
|
|
mlir::Location loc = parser.getEncodedSourceLoc(parser.getNameLoc());
|
|
|
|
return getChecked(loc, loc.getContext(), dimension, p);
|
|
}
|
|
|
|
void CleartextType::print(mlir::AsmPrinter &p) const {
|
|
p << "<";
|
|
if (getP() == -1)
|
|
p << "_";
|
|
else
|
|
p << getP();
|
|
p << ">";
|
|
}
|
|
|
|
mlir::Type CleartextType::parse(mlir::AsmParser &parser) {
|
|
if (parser.parseLess())
|
|
return mlir::Type();
|
|
|
|
int p = -1;
|
|
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(p))
|
|
return mlir::Type();
|
|
if (parser.parseGreater())
|
|
return mlir::Type();
|
|
|
|
Location loc = parser.getEncodedSourceLoc(parser.getNameLoc());
|
|
|
|
return getChecked(loc, loc.getContext(), p);
|
|
}
|
|
|
|
void PlaintextType::print(mlir::AsmPrinter &p) const {
|
|
p << "<";
|
|
if (getP() == -1)
|
|
p << "_";
|
|
else
|
|
p << getP();
|
|
p << ">";
|
|
}
|
|
|
|
mlir::Type PlaintextType::parse(mlir::AsmParser &parser) {
|
|
|
|
if (parser.parseLess())
|
|
return mlir::Type();
|
|
|
|
int p = -1;
|
|
|
|
if (parser.parseOptionalKeyword("_") && parser.parseInteger(p))
|
|
return mlir::Type();
|
|
if (parser.parseGreater())
|
|
return mlir::Type();
|
|
|
|
Location loc = parser.getEncodedSourceLoc(parser.getNameLoc());
|
|
|
|
return getChecked(loc, loc.getContext(), p);
|
|
}
|