Segfault happens with examples/main.c

[gigony]

Hi @yamaguchi1024,

While trying to execute the example code, I found that there is a regression due to the following PR:

• Rework the Scheduling Example #468

The order of the parameters has changed from C, A, B (by your original code) to A, C, B (by PR 468), but the main.c file wasn't updated.

@proc
def rank_k_reduce_6x16(
    K: size,
    A: f32[6, K] @ DRAM,
    C: f32[6, 16] @ DRAM,
    B: f32[K, 16] @ DRAM,
):

I think the following change is needed:

diff --git a/examples/main.c b/examples/main.c
index 5051c42..fbbed91 100644
--- a/examples/main.c
+++ b/examples/main.c
@@ -34,7 +34,7 @@ int main() {
   // Calling original matmul
   start = clock();
   for (int i = 0; i < 1000; i++)
-    rank_k_reduce_6x16(NULL, K, C, A, B);
+    rank_k_reduce_6x16(NULL, K, A, C, B);
   end = clock();
 
   msec = (end - start) * 1000 / CLOCKS_PER_SEC;
@@ -44,7 +44,7 @@ int main() {
   // Calling scheduled matmul
   start = clock();
   for (int i = 0; i < 1000; i++)
-    rank_k_reduce_6x16_scheduled(NULL, K, C, A, B);
+    rank_k_reduce_6x16_scheduled(NULL, K, A, C, B);
   end = clock();
 
   msec = (end - start) * 1000 / CLOCKS_PER_SEC;

How to reproduce:

cd examples
exocc -o . --stem avx2_matmul x86_matmul.py
gcc -o avx2_matmul -march=native main.c avx2_matmul.c
./avx2_matmul
# [1]    3897329 segmentation fault (core dumped)  ./avx2_matmul

After fixing the issue locally, I can run it:

❯ ./avx2_matmul
Time taken for original matmul: 0 seconds 291 milliseconds
Time taken for scheduled matmul: 0 seconds 47 milliseconds

Thanks!

[yamaguchi1024]

Hi @gigony, thanks for trying out Exo and thanks for creating an issue!! I will fix this issue but feel free to open a PR as well