github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

enhance(compiler): genDynamicRankCall.py: Match clang-format formatting rules

Browse Source
This commit is contained in:
Andi Drebes
2022-03-28 10:41:22 +02:00
committed by Quentin Bourgerie
parent b6cec58549
commit e83c421bce
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
compiler/lib/ServerLib/genDynamicRankCall.py
View File

@@ -19,23 +19,23 @@ print(
namespace concretelang {
namespace serverlib {
TensorData
multi_arity_call_dynamic_rank(void* (*func)(void *...), std::vector<void *> args, size_t rank) {
TensorData multi_arity_call_dynamic_rank(void *(*func)(void *...),
std::vector<void *> args,
size_t rank) {
using concretelang::clientlib::MemRefDescriptor;
constexpr auto convert = concretelang::clientlib::tensorDataFromMemRef;
switch (rank) {""")
for tensor_rank in range(0, 33):
memref_rank = tensor_rank + 1
print(f""" case {tensor_rank}:
{{
auto m = multi_arity_call((MemRefDescriptor<{memref_rank}>(*)(void *...))func, args);
return convert({memref_rank}, m.allocated, m.aligned, m.offset, m.sizes, m.strides);
}}""")
print(f""" case {tensor_rank}: {{
auto m = multi_arity_call((MemRefDescriptor<{memref_rank}> (*)(void *...))func, args);
return convert({memref_rank}, m.allocated, m.aligned, m.offset, m.sizes, m.strides);
}}""")
print("""
default:
assert(false);
default:
assert(false);
}
}""")