mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 20:25:34 -05:00
f1161f7f6d
Quick fix due to ordering of includes, had to add #include <mlir/Transforms/DialectConversion.h> to include/concretelang/Conversion/Utils/GenericOpTypeConversionPattern.h
99 lines
2.3 KiB
C++
99 lines
2.3 KiB
C++
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
|
|
// Exceptions. See
|
|
// https://github.com/zama-ai/homomorphizer/blob/master/LICENSE.txt for license
|
|
// information.
|
|
|
|
#ifndef CONCRETELANG_SUPPORT_CLIENTPARAMETERS_H_
|
|
#define CONCRETELANG_SUPPORT_CLIENTPARAMETERS_H_
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <mlir/Dialect/LLVMIR/LLVMTypes.h>
|
|
#include <mlir/IR/BuiltinOps.h>
|
|
|
|
#include "concretelang/Support/V0Parameters.h"
|
|
|
|
namespace mlir {
|
|
namespace concretelang {
|
|
|
|
typedef size_t DecompositionLevelCount;
|
|
typedef size_t DecompositionBaseLog;
|
|
typedef size_t PolynomialSize;
|
|
typedef size_t Precision;
|
|
typedef double Variance;
|
|
|
|
typedef uint64_t LweSize;
|
|
typedef uint64_t GlweDimension;
|
|
|
|
typedef std::string LweSecretKeyID;
|
|
struct LweSecretKeyParam {
|
|
LweSize size;
|
|
|
|
void hash(size_t &seed);
|
|
};
|
|
|
|
typedef std::string BootstrapKeyID;
|
|
struct BootstrapKeyParam {
|
|
LweSecretKeyID inputSecretKeyID;
|
|
LweSecretKeyID outputSecretKeyID;
|
|
DecompositionLevelCount level;
|
|
DecompositionBaseLog baseLog;
|
|
GlweDimension glweDimension;
|
|
Variance variance;
|
|
|
|
void hash(size_t &seed);
|
|
};
|
|
|
|
typedef std::string KeyswitchKeyID;
|
|
struct KeyswitchKeyParam {
|
|
LweSecretKeyID inputSecretKeyID;
|
|
LweSecretKeyID outputSecretKeyID;
|
|
DecompositionLevelCount level;
|
|
DecompositionBaseLog baseLog;
|
|
Variance variance;
|
|
|
|
void hash(size_t &seed);
|
|
};
|
|
|
|
struct Encoding {
|
|
Precision precision;
|
|
};
|
|
|
|
struct EncryptionGate {
|
|
LweSecretKeyID secretKeyID;
|
|
Variance variance;
|
|
Encoding encoding;
|
|
};
|
|
|
|
struct CircuitGateShape {
|
|
// Width of the scalar value
|
|
size_t width;
|
|
// Dimensions of the tensor, empty if scalar
|
|
std::vector<int64_t> dimensions;
|
|
// Size of the buffer containing the tensor
|
|
size_t size;
|
|
};
|
|
|
|
struct CircuitGate {
|
|
llvm::Optional<EncryptionGate> encryption;
|
|
CircuitGateShape shape;
|
|
};
|
|
|
|
struct ClientParameters {
|
|
std::map<LweSecretKeyID, LweSecretKeyParam> secretKeys;
|
|
std::map<BootstrapKeyID, BootstrapKeyParam> bootstrapKeys;
|
|
std::map<KeyswitchKeyID, KeyswitchKeyParam> keyswitchKeys;
|
|
std::vector<CircuitGate> inputs;
|
|
std::vector<CircuitGate> outputs;
|
|
size_t hash();
|
|
};
|
|
|
|
llvm::Expected<ClientParameters>
|
|
createClientParametersForV0(V0FHEContext context, llvm::StringRef name,
|
|
mlir::ModuleOp module);
|
|
|
|
} // namespace concretelang
|
|
} // namespace mlir
|
|
|
|
#endif |