Fix context-free output_shape

0fb17f71 · charlie · ff195f97 · 0fb17f71 · 0fb17f71
Commit 0fb17f71 authored Sep 29, 2022 by charlie
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

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

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

No files found.
--- a/src/include/migraphx/operation.hpp
+++ b/src/include/migraphx/operation.hpp
@@ -408,7 +408,8 @@ auto is_context_free_op(rank<1>,
                        const T& x,
                        const shape& output_shape,
                        const std::vector<argument>& input)
-    -> decltype(x.compute(output_shape, input), std::true_type{});
+    -> decltype(x.compute(make_compute_output_shape(pack(x, output_shape, input)), input),
+                std::true_type{});
 template <class T>
 auto is_context_free_op(rank<0>, const T&, const shape&, const std::vector<argument>&)

--- a/tools/include/operation.hpp
+++ b/tools/include/operation.hpp
@@ -408,7 +408,8 @@ auto is_context_free_op(rank<1>,
                        const T& x,
                        const shape& output_shape,
                        const std::vector<argument>& input)
-    -> decltype(x.compute(output_shape, input), std::true_type{});
+    -> decltype(x.compute(make_compute_output_shape(pack(x, output_shape, input)), input),
+                std::true_type{});
 template <class T>
 auto is_context_free_op(rank<0>, const T&, const shape&, const std::vector<argument>&)