generate a matrix product with many temporaries

1 files changed, 20 insertions, 0 deletions

diff --git a/test/monniaux/many_temporaries/matrix_product.py b/test/monniaux/many_temporaries/matrix_product.py
new file mode 100644
index 00000000..911ca92c
--- /dev/null
+++ b/test/monniaux/many_temporaries/matrix_product.py
@@ -0,0 +1,20 @@
+m=4
+n=5
+p=6
+number_type='double'
+with open('matrix_product.c', 'w') as cfile:
+    cfile.write(f'void matrix_product({number_type} c[{m}][{p}], const {number_type} a[{m}][{n}], const {number_type} b[{n}][{p}]) {{\n')
+    for i in range(m):
+        for j in range(n):
+            for k in range(p):
+                cfile.write(f'  const {number_type} p_{i}_{j}_{k} = a[{i}][{j}] * b[{j}][{k}];\n')
+    for i in range(m):
+        for k in range(p):
+            cfile.write(f'  c[{i}][{k}] = ')
+            for j in range(n):
+                if j>0:
+                    cfile.write(' + ')
+                cfile.write(f'p_{i}_{j}_{k}')
+            cfile.write(';\n')
+    cfile.write('}\n')
+