zsk/fix adapative_pool_2d (#852)

* fix adapative_pool_2d --------- Co-authored-by: yangbofun <[email protected]>
DeepLink-org · Jan 16, 2024 · 42ae84b · 42ae84b
1 parent 2cc9690
commit 42ae84b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 17 deletions.
diff --git a/impl/ascend/device_configs.py b/impl/ascend/device_configs.py
@@ -231,14 +231,6 @@
     'adaptive_avg_pool2d': dict(
         name=['adaptive_avg_pool2d'],
         atol=2e-2,
-        tensor_para=dict(
-            args=[
-                {
-                    "ins": ['input'],
-                    "shape": [Skip((3,16,8)), Skip((4,16,12)), Skip((2,144,65,65))],
-                },
-            ]
-        ),
     ),
 
     'adaptive_max_pool2d': dict(

diff --git a/impl/ascend/functions/pool.cpp b/impl/ascend/functions/pool.cpp
@@ -5,7 +5,6 @@
  */
 
 #include "../common/acloprunner.hpp"
-
 namespace impl {
 namespace ascend {
 
@@ -37,14 +36,12 @@ diopiError_t diopiAdaptiveAvgPool2dBackward(diopiContextHandle_t ctx, diopiTenso
         diopiMulInp(ctx, gradInput, gradOutput);
         return diopiSuccess;
     }
-
-    AclOpRunner<1, 1>("AdaptiveAvgPool2dGrad", ctx)
-        .addInput(gradOutput)
-        .setAttr("orig_input_shape",
-                 std::vector<int32_t>{
-                     static_cast<int>(shape.data[0]), static_cast<int>(shape.data[1]), static_cast<int>(shape.data[2]), static_cast<int>(shape.data[3])})
-        .addOutput(gradInput)
-        .run();
+    std::vector<int32_t> shapeVector;
+    shapeVector.reserve(shape.len);
+    for (int i = 0; i < shape.len; ++i) {
+        shapeVector.push_back(static_cast<int>(shape.data[i]));
+    }
+    AclOpRunner<1, 1>("AdaptiveAvgPool2dGrad", ctx).addInput(gradOutput).setAttr("orig_input_shape", shapeVector).addOutput(gradInput).run();
     return diopiSuccess;
 }