mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04: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.
26 lines
748 B
C++
26 lines
748 B
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.
|
|
|
|
#ifndef CONCRETELANG_SERVERLIB_DYNAMIC_RANK_CALL_H
|
|
#define CONCRETELANG_SERVERLIB_DYNAMIC_RANK_CALL_H
|
|
|
|
#include <vector>
|
|
|
|
#include "concretelang/ClientLib/Types.h"
|
|
|
|
namespace concretelang {
|
|
namespace serverlib {
|
|
|
|
using concretelang::clientlib::TensorData;
|
|
|
|
TensorData multi_arity_call_dynamic_rank(void *(*func)(void *...),
|
|
std::vector<void *> args, size_t rank,
|
|
size_t element_width, bool is_signed);
|
|
|
|
} // namespace serverlib
|
|
} // namespace concretelang
|
|
|
|
#endif
|