Commit 55b09edb authored by charlie's avatar charlie
Browse files

Fix bugs

parent 931fe619
...@@ -128,8 +128,8 @@ struct broadcast ...@@ -128,8 +128,8 @@ struct broadcast
{ {
MIGRAPHX_THROW("BROADCAST_2in: s0 length doesn't match with static s1 axis " MIGRAPHX_THROW("BROADCAST_2in: s0 length doesn't match with static s1 axis "
"dimension length (" + "dimension length (" +
migraphx::to_string(s0.dyn_dims()[0]) + migraphx::to_string(s0.lens()[0]) +
" != " + migraphx::to_string(s1.dyn_dims()[axis]) + ")"); " != " + migraphx::to_string(s1.lens()[axis]) + ")");
} }
std::vector<size_t> bcast_strides(s1.ndim(), 0); std::vector<size_t> bcast_strides(s1.ndim(), 0);
std::copy(s0.strides().begin(), s0.strides().end(), bcast_strides.begin() + axis); std::copy(s0.strides().begin(), s0.strides().end(), bcast_strides.begin() + axis);
......
...@@ -58,10 +58,10 @@ struct parse_binary_op : op_parser<parse_binary_op> ...@@ -58,10 +58,10 @@ struct parse_binary_op : op_parser<parse_binary_op>
if(broadcasted != 0) if(broadcasted != 0)
{ {
if(std::any_of( if(std::any_of(
args.cbegin(), args.cend(), [](auto a) { a->get_shape().dynamic(); })) args.cbegin(), args.cend(), [](auto a) { return a->get_shape().dynamic(); }))
{ {
MIGRAPHX_THROW( MIGRAPHX_THROW(
"binary op broadcast attribute not supported for dynamic input shapes"); "Binary op broadcast attribute not supported for dynamic input shapes");
} }
uint64_t axis = parser.parse_value(info.attributes.at("axis")).at<uint64_t>(); uint64_t axis = parser.parse_value(info.attributes.at("axis")).at<uint64_t>();
auto l = info.add_instruction( auto l = info.add_instruction(
......
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