From 2aac31302f8e28cbff4159f50c72b762f2c954c0 Mon Sep 17 00:00:00 2001 From: Paul Date: Fri, 20 Dec 2024 11:10:19 -0600 Subject: [PATCH] Propagate layout with nonpacked inputs --- src/include/migraphx/op/binary.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/include/migraphx/op/binary.hpp b/src/include/migraphx/op/binary.hpp index 782c75ac45b..73123d54ecb 100644 --- a/src/include/migraphx/op/binary.hpp +++ b/src/include/migraphx/op/binary.hpp @@ -88,7 +88,7 @@ struct binary : op_name } else { - return {s0.type(), s0.lens()}; + return shape::from_permutation(s0.type(), s0.lens(), find_permutation({s0, s1})); } }