Fix transposed padding (#3342)

ROCm · Aug 16, 2024 · 05b2ff4 · 05b2ff4
1 parent b8f69be
commit 05b2ff4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/include/migraphx/op/pad.hpp b/src/include/migraphx/op/pad.hpp
@@ -79,8 +79,7 @@ struct pad
             {
                 rdims[i] += pads[i] + pads[i + num_dims];
             }
-            shape s{s0.type(), rdims};
-            return s;
+            return s0.with_lens(rdims);
         }
     }
 

diff --git a/src/targets/gpu/jit/pad.cpp b/src/targets/gpu/jit/pad.cpp
@@ -79,7 +79,7 @@ struct pad_compiler : compiler<pad_compiler>
 
         auto vinputs = inputs;
         vinputs.push_back(inputs.front().with_lens(offset_lens));
-        auto rinputs = reduce_dims(vinputs);
+        auto rinputs = reduce_dims(normalize_permutation(vinputs));
 
         auto rinput_lens  = rinputs.front().lens();
         auto roffset_lens = rinputs.back().lens();