Skip to content

Commit

Permalink
Finish alloc kernel
Browse files Browse the repository at this point in the history
  • Loading branch information
JosseVanDelm committed Dec 6, 2024
1 parent 2ac27f0 commit 3f141b6
Showing 1 changed file with 28 additions and 1 deletion.
29 changes: 28 additions & 1 deletion kernels/alloc/Snakefile
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,23 @@ from util.snake.configs import get_snax_mac_config

config = get_snax_mac_config()

config["snaxoptflags"] = ",".join(
[
"dispatch-kernels",
"set-memory-space",
"set-memory-layout",
"realize-memref-casts",
"reuse-memref-allocs",
"insert-sync-barrier",
"dispatch-regions",
"linalg-to-library-call",
"snax-copy-to-dma",
"memref-to-snax",
"snax-to-func",
"clear-memory-space",
]
)


module default_rules:
snakefile:
Expand All @@ -10,7 +27,17 @@ module default_rules:
config


use rule * from default_rules as default_*
use rule * from default_rules exclude compile_simple_main as default_*


rule compile_snax_binary:
input:
"func.o",
"main.o",
output:
"func.x",
shell:
"{config[ld]} {config[ldflags]} {input} -o {output}"


rule all:
Expand Down

0 comments on commit 3f141b6

Please sign in to comment.