mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-18 00:21:36 -05:00
e8ef48ffd8
This commit: + Adds support for a protocol which enables inter-op between concrete, tfhe-rs and potentially other contributors to the fhe ecosystem. + Gets rid of hand-made serialization in the compiler, and client/server libs. + Refactors client/server libs to allow more pre/post processing of circuit inputs/outputs. The protocol is supported by a definition in the shape of a capnp file, which defines different types of objects among which: + ProgramInfo object, which is a precise description of a set of fhe circuit coming from the same compilation (understand function type information), and the associated key set. + *Key objects, which represent secret/public keys used to encrypt/execute fhe circuits. + Value object, which represent values that can be transferred between client and server to support calls to fhe circuits. The hand-rolled serialization that was previously used is completely dropped in favor of capnp in the whole codebase. The client/server libs, are refactored to introduce a modular design for pre-post processing. Reading the ProgramInfo file associated with a compilation, the client and server libs assemble a pipeline of transformers (functions) for pre and post processing of values coming in and out of a circuit. This design properly decouples various aspects of the processing, and allows these capabilities to be safely extended. In practice this commit includes the following: + Defines the specification in a concreteprotocol package + Integrate the compilation of this package as a compiler dependency via cmake + Modify the compiler to use the Encodings objects defined in the protocol + Modify the compiler to emit ProgramInfo files as compilation artifact, and gets rid of the bloated ClientParameters. + Introduces a new Common library containing the functionalities shared between the compiler and the client/server libs. + Introduces a functional pre-post processing pipeline to this common library + Modify the client/server libs to support loading ProgramInfo objects, and calling circuits using Value messages. + Drops support of JIT. + Drops support of C-api. + Drops support of Rust bindings. Co-authored-by: Nikita Frolov <nf@mkmks.org>
282 lines
10 KiB
C++
282 lines
10 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/Common/Values.h"
|
|
#include "capnp/common.h"
|
|
#include "capnp/list.h"
|
|
#include "concrete-protocol.capnp.h"
|
|
#include "concretelang/Common/Error.h"
|
|
#include "concretelang/Common/Protocol.h"
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <stdlib.h>
|
|
#include <string>
|
|
|
|
using concretelang::error::Result;
|
|
using concretelang::error::StringError;
|
|
using concretelang::protocol::dimensionsToProtoShape;
|
|
using concretelang::protocol::Message;
|
|
using concretelang::protocol::protoPayloadToVector;
|
|
using concretelang::protocol::protoShapeToDimensions;
|
|
using concretelang::protocol::vectorToProtoPayload;
|
|
|
|
namespace concretelang {
|
|
namespace values {
|
|
|
|
Value Value::fromRawTransportValue(TransportValue transportVal) {
|
|
Value output;
|
|
auto integerPrecision =
|
|
transportVal.asReader().getRawInfo().getIntegerPrecision();
|
|
auto isSigned = transportVal.asReader().getRawInfo().getIsSigned();
|
|
auto dimensions =
|
|
protoShapeToDimensions(transportVal.asReader().getRawInfo().getShape());
|
|
auto data = transportVal.asReader().getPayload();
|
|
if (integerPrecision == 8 && isSigned) {
|
|
auto values = protoPayloadToVector<int8_t>(data);
|
|
output.inner = Tensor<int8_t>{values, dimensions};
|
|
} else if (integerPrecision == 16 && isSigned) {
|
|
auto values = protoPayloadToVector<int16_t>(data);
|
|
output.inner = Tensor<int16_t>{values, dimensions};
|
|
} else if (integerPrecision == 32 && isSigned) {
|
|
auto values = protoPayloadToVector<int32_t>(data);
|
|
output.inner = Tensor<int32_t>{values, dimensions};
|
|
} else if (integerPrecision == 64 && isSigned) {
|
|
auto values = protoPayloadToVector<int64_t>(data);
|
|
output.inner = Tensor<int64_t>{values, dimensions};
|
|
} else if (integerPrecision == 8 && !isSigned) {
|
|
auto values = protoPayloadToVector<uint8_t>(data);
|
|
output.inner = Tensor<uint8_t>{values, dimensions};
|
|
} else if (integerPrecision == 16 && !isSigned) {
|
|
auto values = protoPayloadToVector<uint16_t>(data);
|
|
output.inner = Tensor<uint16_t>{values, dimensions};
|
|
} else if (integerPrecision == 32 && !isSigned) {
|
|
auto values = protoPayloadToVector<uint32_t>(data);
|
|
output.inner = Tensor<uint32_t>{values, dimensions};
|
|
} else if (integerPrecision == 64 && !isSigned) {
|
|
auto values = protoPayloadToVector<uint64_t>(data);
|
|
output.inner = Tensor<uint64_t>{values, dimensions};
|
|
} else {
|
|
assert(false);
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
TransportValue Value::intoRawTransportValue() const {
|
|
auto output = Message<concreteprotocol::Value>();
|
|
auto rawInfo = output.asBuilder().initRawInfo();
|
|
rawInfo.setShape(intoProtoShape().asReader());
|
|
rawInfo.setIntegerPrecision(getIntegerPrecision());
|
|
rawInfo.setIsSigned(isSigned());
|
|
output.asBuilder().setPayload(intoProtoPayload().asReader());
|
|
return output;
|
|
}
|
|
|
|
uint32_t Value::getIntegerPrecision() const {
|
|
if (hasElementType<uint8_t>() || hasElementType<int8_t>()) {
|
|
return 8;
|
|
} else if (hasElementType<uint16_t>() || hasElementType<int16_t>()) {
|
|
return 16;
|
|
} else if (hasElementType<uint32_t>() || hasElementType<int32_t>()) {
|
|
return 32;
|
|
} else if (hasElementType<uint64_t>() || hasElementType<int64_t>()) {
|
|
return 64;
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
bool Value::isSigned() const {
|
|
|
|
if (hasElementType<uint8_t>() || hasElementType<uint16_t>() ||
|
|
hasElementType<uint32_t>() || hasElementType<uint64_t>()) {
|
|
return false;
|
|
} else if (hasElementType<int8_t>() || hasElementType<int16_t>() ||
|
|
hasElementType<int32_t>() || hasElementType<int64_t>()) {
|
|
return true;
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
Message<concreteprotocol::Payload> Value::intoProtoPayload() const {
|
|
if (hasElementType<uint8_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<uint8_t>>(inner).values);
|
|
} else if (hasElementType<uint16_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<uint16_t>>(inner).values);
|
|
} else if (hasElementType<uint32_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<uint32_t>>(inner).values);
|
|
} else if (hasElementType<uint64_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<uint64_t>>(inner).values);
|
|
} else if (hasElementType<int8_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<int8_t>>(inner).values);
|
|
} else if (hasElementType<int16_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<int16_t>>(inner).values);
|
|
} else if (hasElementType<int32_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<int32_t>>(inner).values);
|
|
} else if (hasElementType<int64_t>()) {
|
|
return vectorToProtoPayload(std::get<Tensor<int64_t>>(inner).values);
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
Message<concreteprotocol::Shape> Value::intoProtoShape() const {
|
|
return dimensionsToProtoShape(getDimensions());
|
|
}
|
|
|
|
std::vector<size_t> Value::getDimensions() const {
|
|
if (auto tensor = getTensor<uint8_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<uint16_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<uint32_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<uint64_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<int8_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<int16_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<int32_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else if (auto tensor = getTensor<int64_t>(); tensor) {
|
|
return tensor.value().dimensions;
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
size_t Value::getLength() const {
|
|
if (auto tensor = getTensor<uint8_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<uint16_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<uint32_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<uint64_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<int8_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<int16_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<int32_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else if (auto tensor = getTensor<int64_t>(); tensor) {
|
|
return tensor.value().values.size();
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
bool Value::isCompatibleWithShape(
|
|
const Message<concreteprotocol::Shape> &shape) const {
|
|
auto dimensions = getDimensions();
|
|
if ((uint32_t)shape.asReader().getDimensions().size() != dimensions.size()) {
|
|
return false;
|
|
}
|
|
for (uint32_t i = 0; i < dimensions.size(); i++) {
|
|
if (shape.asReader().getDimensions()[i] != dimensions[i]) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool Value::operator==(const Value &b) const {
|
|
if (auto tensor = getTensor<uint8_t>(); tensor) {
|
|
return tensor == b.getTensor<uint8_t>();
|
|
} else if (auto tensor = getTensor<uint16_t>(); tensor) {
|
|
return tensor == b.getTensor<uint16_t>();
|
|
} else if (auto tensor = getTensor<uint32_t>(); tensor) {
|
|
return tensor == b.getTensor<uint32_t>();
|
|
} else if (auto tensor = getTensor<uint64_t>(); tensor) {
|
|
return tensor == b.getTensor<uint64_t>();
|
|
} else if (auto tensor = getTensor<int8_t>(); tensor) {
|
|
return tensor == b.getTensor<int8_t>();
|
|
} else if (auto tensor = getTensor<int16_t>(); tensor) {
|
|
return tensor == b.getTensor<int16_t>();
|
|
} else if (auto tensor = getTensor<int32_t>(); tensor) {
|
|
return tensor == b.getTensor<int32_t>();
|
|
} else if (auto tensor = getTensor<int64_t>(); tensor) {
|
|
return tensor == b.getTensor<int64_t>();
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
bool Value::isScalar() const {
|
|
if (auto tensor = getTensor<int8_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<int16_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<int32_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<int64_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<uint8_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<uint16_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<uint32_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else if (auto tensor = getTensor<uint64_t>(); tensor) {
|
|
return tensor.value().isScalar();
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
Value Value::toUnsigned() const {
|
|
if (!this->isSigned()) {
|
|
return *this;
|
|
} else if (auto tensor = getTensor<int8_t>(); tensor) {
|
|
return Value((Tensor<uint8_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<int16_t>(); tensor) {
|
|
return Value((Tensor<uint16_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<int32_t>(); tensor) {
|
|
return Value((Tensor<uint32_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<int64_t>(); tensor) {
|
|
return Value((Tensor<uint8_t>)tensor.value());
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
Value Value::toSigned() const {
|
|
if (!this->isSigned()) {
|
|
return *this;
|
|
} else if (auto tensor = getTensor<uint8_t>(); tensor) {
|
|
return Value((Tensor<int8_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<uint16_t>(); tensor) {
|
|
return Value((Tensor<int16_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<uint32_t>(); tensor) {
|
|
return Value((Tensor<int32_t>)tensor.value());
|
|
} else if (auto tensor = getTensor<uint64_t>(); tensor) {
|
|
return Value((Tensor<int8_t>)tensor.value());
|
|
} else {
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
size_t getCorrespondingPrecision(size_t originalPrecision) {
|
|
if (originalPrecision <= 8) {
|
|
return 8;
|
|
}
|
|
if (originalPrecision <= 16) {
|
|
return 16;
|
|
}
|
|
if (originalPrecision <= 32) {
|
|
return 32;
|
|
}
|
|
if (originalPrecision <= 64) {
|
|
return 64;
|
|
}
|
|
assert(false);
|
|
}
|
|
|
|
} // namespace values
|
|
} // namespace concretelang
|