Ensure unique module name for MLIR standalone ops (#2360)

d1abf06f · Paul Fultz II · GitHub · 7604ecf5 · d1abf06f
Unverified Commit d1abf06f authored Oct 24, 2023 by Paul Fultz II Committed by GitHub Oct 24, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/targets/gpu/fuse_mlir.cpp src/targets/gpu/fuse_mlir.cpp +2 -1

No files found.
--- a/src/targets/gpu/fuse_mlir.cpp
+++ b/src/targets/gpu/fuse_mlir.cpp
@@ -361,7 +361,8 @@ struct find_mlir_standalone_op
            return;

        static size_t counter = 0;
-        module_ref mm         = mpm.create_module("mlir_" + std::to_string(counter++));
+        module_ref mm =
+            mpm.create_module("mlir_" + conv_based_op->name() + std::to_string(counter++));
        mm->set_bypass();
        auto [anchor_op, top_inputs] = fuse_input_ops_and_gemm_based_op(mm, conv_based_op);
        mm->add_return({anchor_op});