Propagate layout with nonpacked inputs

ROCm · Dec 20, 2024 · 2aac313 · 2aac313
1 parent a678b48
commit 2aac313
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/include/migraphx/op/binary.hpp b/src/include/migraphx/op/binary.hpp
@@ -88,7 +88,7 @@ struct binary : op_name<Derived>
         }
         else
         {
-            return {s0.type(), s0.lens()};
+            return shape::from_permutation(s0.type(), s0.lens(), find_permutation({s0, s1}));
         }
     }