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

fix(frontend-python): broadcast tlu input on multi tlus

Browse Source
This commit is contained in:
Umut
2023-05-11 10:04:45 +02:00
parent 4c88557dee
commit d99436e098
2 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
frontends/concrete-python/tests/execution/test_others.py
View File

@@ -914,3 +914,33 @@ def test_others_bad_univariate(helpers):
"Function bad_univariate cannot be used with fhe.univariate",
str(excinfo.value),
)
def test_dynamic_indexing_hack(helpers):
"""
Test dynamic indexing using basic operators.
"""
@fhe.compiler({"array": "encrypted", "index": "encrypted"})
def function(array, index):
all_indices = np.arange(array.size)
index_selection = index == all_indices
selection_and_zeros = array * index_selection
selection = np.sum(selection_and_zeros)
return selection
inputset = [
(
np.random.randint(0, 16, size=(4,)),
np.random.randint(0, 4, size=()),
)
for _ in range(100)
]
circuit = function.compile(inputset, helpers.configuration())
sample = np.random.randint(0, 16, size=(4,))
helpers.check_execution(circuit, function, [sample, 0], retries=3)
helpers.check_execution(circuit, function, [sample, 1], retries=3)
helpers.check_execution(circuit, function, [sample, 2], retries=3)
helpers.check_execution(circuit, function, [sample, 3], retries=3)