feat: make get_printable_graph give correct info for np.dot

closes #204
This commit is contained in:
Benoit Chevallier-Mames
2021-09-03 09:47:34 +02:00
committed by Benoit Chevallier
parent 6b6aa7ee4e
commit 150d33ba48
6 changed files with 53 additions and 23 deletions

View File

@@ -213,9 +213,12 @@ def test_fail_compile(function, input_ranges, list_of_arg_names):
(4,),
# Remark that, when you do the dot of tensors of 4 values between 0 and 3,
# you can get a maximal value of 4*3*3 = 36, ie something on 6 bits
"%0 = x # Integer<unsigned, 6 bits>"
"\n%1 = y # Integer<unsigned, 6 bits>"
"\n%2 = Dot(0, 1) # Integer<unsigned, 6 bits>"
"%0 = x "
"# EncryptedTensor<Integer<unsigned, 6 bits>, shape=(4,)>"
"\n%1 = y "
"# EncryptedTensor<Integer<unsigned, 6 bits>, shape=(4,)>"
"\n%2 = Dot(0, 1) "
"# EncryptedScalar<Integer<unsigned, 6 bits>>"
"\nreturn(%2)\n",
),
# pylint: enable=unnecessary-lambda