Actually test more asm functions

mit-plv · Mar 11, 2025 · fef98f1 · fef98f1
1 parent 7aa277a
commit fef98f1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fiat-amd64/gentest.py b/fiat-amd64/gentest.py
@@ -46,7 +46,7 @@ def removeprefix(s, prefix):
 
 asm_op_names = OrderedDict()
 
-regex = re.compile(r'fiat_(?P<name>[^_]+(_(solinas|montgomery|dettman))?)_(?P<op>(carry_)?(square|mul))')
+regex = re.compile(r'fiat_(?P<name>[^_]+(_(solinas|montgomery|dettman))?)_(?P<op>(carry_)?(square|mul|from_bytes|to_bytes|add|sub|opp))')
 for dirname in directories:
     m = regex.match(os.path.basename(dirname))
     if m:
@@ -74,6 +74,9 @@ def asm_op_names_key(val):
         kind = 3
         n, prime = solinasprimes[name]
 
+    else:
+        assert False, name
+
     return (kind, n, prime, op, name, fnames)
 
 def is_small(val):