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`.
248 lines
9.5 KiB
C++
248 lines
9.5 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.
|
|
|
|
// generated: see genDynamicRandCall.py
|
|
|
|
#include <cassert>
|
|
#include <vector>
|
|
|
|
#include "concretelang/ClientLib/Types.h"
|
|
#include "concretelang/ServerLib/DynamicArityCall.h"
|
|
#include "concretelang/ServerLib/ServerLambda.h"
|
|
|
|
namespace concretelang {
|
|
namespace serverlib {
|
|
|
|
/// Helper class template that yields an unsigned integer type given a
|
|
/// size in bytes
|
|
template <std::size_t size> struct int_type_of_size {};
|
|
template <> struct int_type_of_size<4> { typedef uint32_t type; };
|
|
template <> struct int_type_of_size<8> { typedef uint64_t type; };
|
|
|
|
/// Converts one function pointer into another
|
|
// TODO: Not sure this is valid in all implementations / on all
|
|
// architectures
|
|
template <typename FnDstT, typename FnSrcT> FnDstT convert_fnptr(FnSrcT src) {
|
|
static_assert(sizeof(FnDstT) == sizeof(FnSrcT),
|
|
"Size of function types must match");
|
|
using inttype = typename int_type_of_size<sizeof(FnDstT)>::type;
|
|
inttype raw = reinterpret_cast<inttype>(src);
|
|
return reinterpret_cast<FnDstT>(raw);
|
|
}
|
|
|
|
ScalarOrTensorData multi_arity_call_dynamic_rank(void *(*func)(void *...),
|
|
std::vector<void *> args,
|
|
size_t rank,
|
|
size_t element_width,
|
|
bool is_signed) {
|
|
using concretelang::clientlib::MemRefDescriptor;
|
|
constexpr auto convert = concretelang::clientlib::tensorDataFromMemRef;
|
|
switch (rank) {
|
|
case 0: {
|
|
auto m =
|
|
multi_arity_call(convert_fnptr<uint64_t (*)(void *...)>(func), args);
|
|
return concretelang::clientlib::ScalarData(m, is_signed, element_width);
|
|
}
|
|
case 1: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<1> (*)(void *...)>(func), args);
|
|
return convert(1, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 2: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<2> (*)(void *...)>(func), args);
|
|
return convert(2, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 3: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<3> (*)(void *...)>(func), args);
|
|
return convert(3, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 4: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<4> (*)(void *...)>(func), args);
|
|
return convert(4, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 5: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<5> (*)(void *...)>(func), args);
|
|
return convert(5, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 6: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<6> (*)(void *...)>(func), args);
|
|
return convert(6, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 7: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<7> (*)(void *...)>(func), args);
|
|
return convert(7, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 8: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<8> (*)(void *...)>(func), args);
|
|
return convert(8, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 9: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<9> (*)(void *...)>(func), args);
|
|
return convert(9, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 10: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<10> (*)(void *...)>(func), args);
|
|
return convert(10, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 11: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<11> (*)(void *...)>(func), args);
|
|
return convert(11, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 12: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<12> (*)(void *...)>(func), args);
|
|
return convert(12, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 13: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<13> (*)(void *...)>(func), args);
|
|
return convert(13, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 14: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<14> (*)(void *...)>(func), args);
|
|
return convert(14, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 15: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<15> (*)(void *...)>(func), args);
|
|
return convert(15, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 16: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<16> (*)(void *...)>(func), args);
|
|
return convert(16, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 17: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<17> (*)(void *...)>(func), args);
|
|
return convert(17, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 18: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<18> (*)(void *...)>(func), args);
|
|
return convert(18, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 19: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<19> (*)(void *...)>(func), args);
|
|
return convert(19, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 20: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<20> (*)(void *...)>(func), args);
|
|
return convert(20, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 21: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<21> (*)(void *...)>(func), args);
|
|
return convert(21, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 22: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<22> (*)(void *...)>(func), args);
|
|
return convert(22, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 23: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<23> (*)(void *...)>(func), args);
|
|
return convert(23, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 24: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<24> (*)(void *...)>(func), args);
|
|
return convert(24, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 25: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<25> (*)(void *...)>(func), args);
|
|
return convert(25, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 26: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<26> (*)(void *...)>(func), args);
|
|
return convert(26, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 27: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<27> (*)(void *...)>(func), args);
|
|
return convert(27, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 28: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<28> (*)(void *...)>(func), args);
|
|
return convert(28, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 29: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<29> (*)(void *...)>(func), args);
|
|
return convert(29, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 30: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<30> (*)(void *...)>(func), args);
|
|
return convert(30, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 31: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<31> (*)(void *...)>(func), args);
|
|
return convert(31, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
case 32: {
|
|
auto m = multi_arity_call(
|
|
convert_fnptr<MemRefDescriptor<32> (*)(void *...)>(func), args);
|
|
return convert(32, element_width, is_signed, m.allocated, m.aligned,
|
|
m.offset, m.sizes, m.strides);
|
|
}
|
|
|
|
default:
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
} // namespace serverlib
|
|
} // namespace concretelang
|