Check dynamic() on shape when using dyn_output

07c05efb · charlie · da5b6fef · 07c05efb · 07c05efb
Commit 07c05efb authored Sep 27, 2022 by charlie
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

src/include/migraphx/operation.hpp src/include/migraphx/operation.hpp +3 -1

tools/include/operation.hpp tools/include/operation.hpp +3 -1

No files found.
--- a/src/include/migraphx/operation.hpp
+++ b/src/include/migraphx/operation.hpp
@@ -116,7 +116,9 @@ struct compute_output_shape
    operator dyn_output() const
    {
        return ins_inputs([](const auto& x, shape ins_shape, const std::vector<argument>& inputs) {
-            return dyn_output{ins_shape, compute_shape(x, to_shapes(inputs))};
+            if(ins_shape.dynamic())
+                return dyn_output{ins_shape, compute_shape(x, to_shapes(inputs))};
+            return dyn_output{ins_shape, ins_shape};
        });
    }

--- a/tools/include/operation.hpp
+++ b/tools/include/operation.hpp
@@ -116,7 +116,9 @@ struct compute_output_shape
    operator dyn_output() const
    {
        return ins_inputs([](const auto& x, shape ins_shape, const std::vector<argument>& inputs) {
-            return dyn_output{ins_shape, compute_shape(x, to_shapes(inputs))};
+            if(ins_shape.dynamic())
+                return dyn_output{ins_shape, compute_shape(x, to_shapes(inputs))};
+            return dyn_output{ins_shape, ins_shape};
        });
    }