Snakemake: Add streamer alu (#317)

* Add snax alu config

* Add streamer-alu snakefile

* Makefile

* Move streamer_alu to snakemake ci

.github/workflows/build-run-kernel-snake.yml

@@ -20,4 +20,4 @@ jobs:
         working-directory: kernels/${{ matrix.kernel }}
     strategy:
       matrix:
-        kernel: [alloc, simple_copy, transform_copy, gemm, rescale, gemmini]
+        kernel: [alloc, simple_copy, transform_copy, gemm, rescale, gemmini, streamer_alu]

.github/workflows/build-run-kernel.yml

@@ -20,4 +20,4 @@ jobs:
         working-directory: kernels/${{ matrix.kernel }}
     strategy:
       matrix:
-        kernel: [streamer_alu, tiled_add, streamer_matmul]
+        kernel: [tiled_add, streamer_matmul]

kernels/streamer_alu/Snakefile

@@ -0,0 +1,79 @@
+from util.snake.configs import get_snax_alu_config
+
+config = get_snax_alu_config()
+config["snaxoptflags"] = ",".join(
+    [
+        "insert-accfg-op{accelerator=snax_alu}",
+        "convert-linalg-to-kernel",
+        "dispatch-kernels",
+        "test-insert-debugs",
+        "set-memory-space",
+        "set-memory-layout",
+        "test-insert-debugs",
+        "realize-memref-casts",
+        "reuse-memref-allocs",
+        "insert-sync-barrier",
+        "dispatch-regions",
+        "convert-linalg-to-stream",
+        "convert-stream-to-snax-stream",
+        "convert-linalg-to-accfg",
+        "convert-accfg-to-csr",
+        "snax-copy-to-dma",
+        "memref-to-snax",
+        "snax-to-func",
+        "test-debug-to-func",
+        "clear-memory-space",
+    ]
+)
+
+
+module default_rules:
+    snakefile:
+        "../../util/snake/default_rules.smk"
+    config:
+        config
+
+
+use rule * from default_rules as default_*
+
+
+# Rules
+rule all:
+    input:
+        "streamer_add.x",
+        "streamer_add_stream.x",
+    run:
+        for item in input:
+            shell("{config[vltsim]} {item}")
+
+
+from gendata import create_data_files
+
+
+rule generate_data:
+    output:
+        "data.c",
+        "data.h",
+    run:
+        create_data_files()
+
+
+rule compile_simple_main:
+    input:
+        "main.c",
+        "data.o",
+    output:
+        temp("main.o"),
+    shell:
+        "{config[cc]} {config[cflags]} -c {input} -o {output}"
+
+
+rule link_snax_binary:
+    input:
+        "{file}.o",
+        "data.o",
+        "main.o",
+    output:
+        "{file}.x",
+    shell:
+        "{config[ld]} {config[ldflags]} {input} -o {output}"

kernels/streamer_alu/gendata.py

@@ -2,7 +2,8 @@
 
 from util.gendata import create_data, create_header
 
-if __name__ == "__main__":
+
+def create_data_files():
     # Reset random seed for reproducible behavior
     low_bound = 0
     high_bound = 100

util/snake/configs.py

@@ -33,3 +33,14 @@ def get_snax_gemmx_config():
         + "/snax-kul-cluster-mixed-narrow-wide-rtl/bin/snitch_cluster.vlt"
     )
     return config
+
+
+def get_snax_alu_config():
+    # use CONDA_PREFIX to access pixi env
+    snax_utils_path = os.environ["CONDA_PREFIX"] + "/snax-utils"
+    snitch_sw_path = snax_utils_path + "/snax-alu"
+    config = {}
+    config.update(get_default_paths())
+    config.update(get_default_flags(snitch_sw_path))
+    config["vltsim"] = snax_utils_path + "/snax-alu-rtl/bin/snitch_cluster.vlt"
+    return config