feat: add allowTransient for ebytes64 and ebytes128

zama-ai · Dec 14, 2024 · 770ebad · 770ebad
1 parent ad6bfac
commit 770ebad
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/codegen/templates.ts b/codegen/templates.ts
@@ -877,6 +877,14 @@ function tfheAclMethods(supportedBits: number[]): string {
       Impl.allowTransient(eaddress.unwrap(value), account);
     }
 
+    function allowTransient(ebytes64 value, address account) internal {
+      Impl.allowTransient(ebytes64.unwrap(value), account);
+    }
+
+    function allowTransient(ebytes128 value, address account) internal {
+      Impl.allowTransient(ebytes128.unwrap(value), account);
+    }
+
     function allowTransient(ebytes256 value, address account) internal {
       Impl.allowTransient(ebytes256.unwrap(value), account);
     }

diff --git a/lib/TFHE.sol b/lib/TFHE.sol
@@ -11033,6 +11033,14 @@ library TFHE {
         Impl.allowTransient(eaddress.unwrap(value), account);
     }
 
+    function allowTransient(ebytes64 value, address account) internal {
+        Impl.allowTransient(ebytes64.unwrap(value), account);
+    }
+
+    function allowTransient(ebytes128 value, address account) internal {
+        Impl.allowTransient(ebytes128.unwrap(value), account);
+    }
+
     function allowTransient(ebytes256 value, address account) internal {
         Impl.allowTransient(ebytes256.unwrap(value), account);
     }