v4.3.1 update. (#2817)

2025-11-27 09:49:30 -05:00
parent 2052fd3885
commit 1de3a576cc
44 changed files with 3316 additions and 510 deletions
@@ -77,6 +77,7 @@ def main():
    # compile the kernel with "--enable-tvm-ffi" option
    compiled_add_one = cute.compile(add_one, a_cute, b_cute, options="--enable-tvm-ffi")

+    os.makedirs("./build", exist_ok=True)
    object_file_path = "./build/add_one.o"
    lib_path = "./build/add_one.so"
    compiled_add_one.export_to_c(object_file_path, function_name="add_one")