Use func dialect

a4d40fd0 · Paul · 34a7c072 · a4d40fd0
Commit a4d40fd0 authored May 18, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/targets/gpu/mlir.cpp src/targets/gpu/mlir.cpp +2 -2

No files found.
--- a/src/targets/gpu/mlir.cpp
+++ b/src/targets/gpu/mlir.cpp
@@ -402,9 +402,9 @@ struct mlir_program
        MlirBlock result   = fbody.get();
        mlirRegionAppendOwnedBlock(region.get(), fbody.release());

-        auto ops = create_operation_state("builtin.func");
+        auto ops = create_operation_state("func.func");
        ops.add_attributes(
-            {{"type", make_function_type(inputs, outputs)}, {"sym_name", std::string("main")}});
+            {{"function_type", make_function_type(inputs, outputs)}, {"sym_name", std::string("main")}, {"kernel", std::string("mixr")}});
        ops.add_region(std::move(region));
        insert(body, std::move(ops));