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

feat: update compile_torch_model to return compiled quantized module

Browse Source 
closes #898
This commit is contained in:
Arthur Meyre
2021-12-02 16:17:52 +01:00
parent bf70396340
commit c7255dfd66
5 changed files with 109 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/quantization/test_compilation.py
View File

@@ -85,7 +85,7 @@ def test_quantized_module_compilation(
homomorphic_predictions = homomorphic_predictions.reshape(dequant_predictions.shape)
# Make sure homomorphic_predictions are the same as dequant_predictions
if numpy.isclose(homomorphic_predictions.ravel(), dequant_predictions.ravel()).all():
if numpy.isclose(homomorphic_predictions, dequant_predictions).all():
return
# Bad computation after nb_tries