fix: add checks for whether or not the inputs and output have the sam…

…e dimensions
moreh-dev · Apr 23, 2024 · 7e69239 · 7e69239
1 parent cf0aa0f
commit 7e69239
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/include/miopen/pad_constant/problem_description.hpp b/src/include/miopen/pad_constant/problem_description.hpp
@@ -45,8 +45,18 @@ struct ProblemDescription : ProblemDescriptionBase
 
     NetworkConfig MakeNetworkConfig() const override;
 
-    bool IsSameType() const {
-        if (xDesc.GetType() == yDesc.GetType())
+    bool IsSameType() const
+    {
+        if(xDesc.GetType() == yDesc.GetType())
+            return true;
+        return false;
+    }
+
+    bool IsSameShape() const
+    {
+        auto xSize = xDesc.GetSize();
+        auto ySize = yDesc.GetSize();
+        if(xSize == ySize)
             return true;
         return false;
     }

diff --git a/src/solver/pad_constant/pad_constant_fwd_contiguous.cpp b/src/solver/pad_constant/pad_constant_fwd_contiguous.cpp
@@ -43,6 +43,8 @@ bool PadConstantFwdContiguous::IsApplicable(
 {
     if(!problem.IsSameType())
         return false;
+    if(!problem.IsSameShape())
+        return false;
 
     return true;
 }