Throw an error when mlir's outpus shape is non-standard (#2150)

f149b619 · Paul Fultz II · GitHub · 752f13cf · f149b619
Unverified Commit f149b619 authored Sep 14, 2023 by Paul Fultz II Committed by GitHub Sep 14, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

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

No files found.
--- a/src/targets/gpu/fuse_mlir.cpp
+++ b/src/targets/gpu/fuse_mlir.cpp
@@ -103,7 +103,10 @@ struct mlir_op
            }
            if(ins->name() == "@return")
            {
-                return ins_shapes[ins->inputs().at(0)].with_type(type);
+                auto s = ins_shapes[ins->inputs().at(0)].with_type(type);
+                if(not s.standard())
+                    MIGRAPHX_THROW("MLIR doesnt support non-standard output");
+                return s;
            }
            std::vector<shape> input_shapes;
            input_shapes.resize(ins->inputs().size());