Commit 79a5ff10 authored by Paul's avatar Paul
Browse files

Formatting

parent 06b02add
...@@ -9,11 +9,11 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -9,11 +9,11 @@ inline namespace MIGRAPHX_INLINE_NS {
bool skip_propogate(instruction_ref ins) bool skip_propogate(instruction_ref ins)
{ {
if (ins->name() == "@literal") if(ins->name() == "@literal")
return true; return true;
if (ins->get_shape().broadcasted() and not ins->get_shape().scalar()) if(ins->get_shape().broadcasted() and not ins->get_shape().scalar())
return true; return true;
if (ins->get_shape().scalar() and ins->get_shape().elements() != 1) if(ins->get_shape().scalar() and ins->get_shape().elements() != 1)
return true; return true;
return false; return false;
} }
......
...@@ -29,7 +29,7 @@ struct binary ...@@ -29,7 +29,7 @@ struct binary
{ {
check_shapes{inputs}.has(2).same_type().same_dims(); check_shapes{inputs}.has(2).same_type().same_dims();
const auto& s = inputs.front(); const auto& s = inputs.front();
if (s.scalar() and s.elements() == 1) if(s.scalar() and s.elements() == 1)
return {s.type()}; return {s.type()};
return {s.type(), s.lens()}; return {s.type(), s.lens()};
} }
......
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