Skip to content

GitLab

Commit 0fb17f71 authored by charlie's avatar charlie
Browse files

Fix context-free output_shape

parent ff195f97
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
src/include/migraphx/operation.hpp
View file @ 0fb17f71
......@@ -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>&)
......
tools/include/operation.hpp
View file @ 0fb17f71
......@@ -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>&)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment