mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
a7051c2c9c
This patch adds support for scalar results to the client/server protocol and tests. In addition to `TensorData`, a new type `ScalarData` is added. Previous representations of scalar values using one-dimensional `TensorData` instances have been replaced with proper instantiations of `ScalarData`. The generic use of `TensorData` for scalar and tensor values has been replaced with uses of a new variant `ScalarOrTensorData`, which can either hold an instance of `TensorData` or `ScalarData`.
90 lines
3.0 KiB
C++
90 lines
3.0 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/ClientLib/EncryptedArguments.h"
|
|
#include "concretelang/ClientLib/PublicArguments.h"
|
|
|
|
namespace concretelang {
|
|
namespace clientlib {
|
|
|
|
using StringError = concretelang::error::StringError;
|
|
|
|
outcome::checked<std::unique_ptr<PublicArguments>, StringError>
|
|
EncryptedArguments::exportPublicArguments(ClientParameters clientParameters,
|
|
RuntimeContext runtimeContext) {
|
|
return std::make_unique<PublicArguments>(
|
|
clientParameters, std::move(preparedArgs), std::move(ciphertextBuffers));
|
|
}
|
|
|
|
outcome::checked<void, StringError>
|
|
EncryptedArguments::pushArg(uint64_t arg, KeySet &keySet) {
|
|
OUTCOME_TRYV(checkPushTooManyArgs(keySet));
|
|
auto pos = currentPos++;
|
|
OUTCOME_TRY(CircuitGate input, keySet.clientParameters().input(pos));
|
|
if (input.shape.size != 0) {
|
|
return StringError("argument #") << pos << " is not a scalar";
|
|
}
|
|
if (!input.encryption.hasValue()) {
|
|
// clear scalar: just push the argument
|
|
preparedArgs.push_back((void *)arg);
|
|
return outcome::success();
|
|
}
|
|
|
|
std::vector<int64_t> shape = keySet.clientParameters().bufferShape(input);
|
|
|
|
// Allocate empty
|
|
ciphertextBuffers.emplace_back(
|
|
TensorData(shape, clientlib::EncryptedScalarElementType,
|
|
clientlib::EncryptedScalarElementWidth));
|
|
TensorData &values_and_sizes = ciphertextBuffers.back().getTensor();
|
|
|
|
OUTCOME_TRYV(keySet.encrypt_lwe(
|
|
pos, values_and_sizes.getElementPointer<decrypted_scalar_t>(0), arg));
|
|
// Note: Since we bufferized lwe ciphertext take care of memref calling
|
|
// convention
|
|
// allocated
|
|
preparedArgs.push_back(nullptr);
|
|
// aligned
|
|
preparedArgs.push_back((void *)values_and_sizes.getValuesAsOpaquePointer());
|
|
// offset
|
|
preparedArgs.push_back((void *)0);
|
|
// sizes
|
|
for (auto size : values_and_sizes.getDimensions()) {
|
|
preparedArgs.push_back((void *)size);
|
|
}
|
|
// strides
|
|
int64_t stride = TensorData::getNumElements(shape);
|
|
for (size_t size : values_and_sizes.getDimensions()) {
|
|
stride = (size == 0 ? 0 : (stride / size));
|
|
preparedArgs.push_back((void *)stride);
|
|
}
|
|
|
|
return outcome::success();
|
|
}
|
|
|
|
outcome::checked<void, StringError>
|
|
EncryptedArguments::checkPushTooManyArgs(KeySet &keySet) {
|
|
size_t arity = keySet.numInputs();
|
|
if (currentPos < arity) {
|
|
return outcome::success();
|
|
}
|
|
return StringError("function has arity ")
|
|
<< arity << " but is applied to too many arguments";
|
|
}
|
|
|
|
outcome::checked<void, StringError>
|
|
EncryptedArguments::checkAllArgs(KeySet &keySet) {
|
|
size_t arity = keySet.numInputs();
|
|
if (currentPos == arity) {
|
|
return outcome::success();
|
|
}
|
|
return StringError("function expects ")
|
|
<< arity << " arguments but has been called with " << currentPos
|
|
<< " arguments";
|
|
}
|
|
|
|
} // namespace clientlib
|
|
} // namespace concretelang
|