mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 12:15:09 -05:00
8255d3e190
Returning tensors with elements whose width is not equal to 64 results in garbled data. This commit extends the `TensorData` class used to represent tensors in JIT compilation with support for signed / unsigned elements of 8/16/32 and 64 bits, such that all clear text tensors with up to 64 bits can be represented accurately.
88 lines
2.9 KiB
C++
88 lines
2.9 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(shape, clientlib::EncryptedScalarElementType);
|
|
TensorData &values_and_sizes = ciphertextBuffers.back();
|
|
|
|
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
|