mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
fix(clientlib): Allow tensor with dimensions equals to 0
This commit is contained in:
@@ -39,12 +39,10 @@ struct TensorData {
|
||||
|
||||
inline size_t length() {
|
||||
if (sizes.empty()) {
|
||||
assert(false);
|
||||
return 0;
|
||||
}
|
||||
size_t len = 1;
|
||||
for (auto size : sizes) {
|
||||
assert(size > 0);
|
||||
len *= size;
|
||||
}
|
||||
return len;
|
||||
|
||||
@@ -142,12 +142,12 @@ EncryptedArguments::pushArg(size_t width, const void *data,
|
||||
for (size_t size : values_and_sizes.sizes) {
|
||||
preparedArgs.push_back((void *)size);
|
||||
}
|
||||
|
||||
// Set the stride for each dimension, equal to the product of the
|
||||
// following dimensions.
|
||||
int64_t stride = values_and_sizes.length();
|
||||
// If encrypted +1 set the stride for the lwe size rank
|
||||
for (size_t size : values_and_sizes.sizes) {
|
||||
stride /= size;
|
||||
stride = (size == 0 ? 0 : (stride / size));
|
||||
preparedArgs.push_back((void *)stride);
|
||||
}
|
||||
currentPos++;
|
||||
|
||||
Reference in New Issue
Block a user