Run simulations in parallel

kernels/alloc/Snakefile

@@ -28,14 +28,30 @@ module default_rules:
         config
 
 
-rule all:
-    input:
-        "func.x",
-    shell:
-        "{config[vltsim]} {input[0]}"
+use rule * from default_rules as default_*
 
 
-use rule * from default_rules exclude compile_simple_main as default_*
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
+files = ["func"]
+
+
+rule all:
+    input:
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 rule compile_snax_binary:

kernels/gemm/Snakefile

@@ -39,6 +39,16 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
 rule compile_main:
     input:
         "main.c",
@@ -49,11 +59,17 @@ rule compile_main:
         "{config[cc]} {config[cflags]} -c {input[0]}"
 
 
+files = ["gemm"]
+
+
 rule all:
     input:
-        "gemm.x",
-    shell:
-        "{config[vltsim]} {input[0]}"
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 from gendata import create_data_files

kernels/rescale/Snakefile

@@ -33,6 +33,16 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
 rule link_snax_binary:
     input:
         "rescale.o",
@@ -44,11 +54,17 @@ rule link_snax_binary:
         "{config[ld]} {config[ldflags]} {input} -o {output}"
 
 
+files = ["rescale"]
+
+
 rule all:
     input:
-        "rescale.x",
-    shell:
-        "{config[vltsim]} {input[0]}"
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 from gendata import create_data_files

kernels/simple_copy/Snakefile

@@ -29,12 +29,28 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
+files = ["simple_copy"]
+
+
 # Rules
 rule all:
     input:
-        "simple_copy.x",
-    shell:
-        "{config[vltsim]} {input[0]}"
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 rule compile_main:

kernels/streamer_alu/Snakefile

@@ -39,14 +39,27 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
-# Rules
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
+files = ["streamer_add", "streamer_add_stream"]
+
+
 rule all:
     input:
-        "streamer_add.x",
-        "streamer_add_stream.x",
-    run:
-        for item in input:
-            shell("{config[vltsim]} {item}")
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 from gendata import create_data_files

kernels/streamer_matmul/Snakefile

@@ -39,17 +39,31 @@ module default_rules:
         config
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
 use rule * from default_rules as default_*
 
 
+use rule * from snax_rules as snax_*
+
+
+files = ["quantized_matmul", "tiled_quantized_matmul"]
+
+
 # Rules
 rule all:
     input:
-        "quantized_matmul.x",
-        "tiled_quantized_matmul.x",
-    run:
-        for item in input:
-            shell("{config[vltsim]} {item}")
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 rule generate_quantized_matmul:

kernels/tiled_add/Snakefile

@@ -55,6 +55,16 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
 rule size_mlir:
     input:
         "{file}.mlir",
@@ -123,15 +133,14 @@ COMPILER_OPTS = ["accfgboth", "noaccfgopt"]
 rule all:
     input:
         expand(
-            "{file}_{array_size}_{tile_size}_{compiler_opt}.x",
+            "{file}_{array_size}_{tile_size}_{compiler_opt}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
             file=FILES,
             array_size=ARRAY_SIZES,
             tile_size=TILE_SIZES,
             compiler_opt=COMPILER_OPTS,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
         ),
-    run:
-        for item in input:
-            shell("{config[vltsim]} {item}")
 
 
 from gendata import generate_data

kernels/transform_copy/Snakefile

@@ -28,19 +28,30 @@ module default_rules:
 use rule * from default_rules as default_*
 
 
+module snax_rules:
+    snakefile:
+        "../../util/snake/snax.smk"
+    config:
+        config
+
+
+use rule * from snax_rules as snax_*
+
+
 from gendata import create_files
 
+files = ["transform_copy", "transform_from_none", "transform_from_strided"]
+
 
 # Rules
 rule all:
     input:
-        "transform_copy.x",
-        "transform_from_none.x",
-        "transform_from_strided.x",
-    run:
-        shell("{config[vltsim]} {input[0]} {input[0]}")
-        shell("{config[vltsim]} {input[1]} {input[1]}")
-        shell("{config[vltsim]} {input[2]} {input[2]}")
+        expand(
+            "{file}_trace_chip_{num_chips:02d}_hart_{num_harts:05d}.dasm",
+            file=files,
+            num_chips=range(config["num_chips"]),
+            num_harts=range(config["num_harts"]),
+        ),
 
 
 rule generate_data: