Unpool/Unpool3d kernel support input of int64 indices 易用性提升 (#480)

update torch.nn.functional.max_unpool
PaddlePaddle · Sep 23, 2024 · 46be5a8 · 46be5a8
1 parent 09a5ec1
commit 46be5a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/paconvert/api_mapping.json b/paconvert/api_mapping.json
@@ -12287,7 +12287,7 @@
     "min_input_args": 0
   },
   "torch.nn.functional.max_unpool1d": {
-    "Matcher": "UnpoolMatcher",
+    "Matcher": "GenericMatcher",
     "paddle_api": "paddle.nn.functional.max_unpool1d",
     "args_list": [
       "input",
@@ -12303,7 +12303,7 @@
     "min_input_args": 3
   },
   "torch.nn.functional.max_unpool2d": {
-    "Matcher": "UnpoolMatcher",
+    "Matcher": "GenericMatcher",
     "paddle_api": "paddle.nn.functional.max_unpool2d",
     "args_list": [
       "input",
@@ -12319,7 +12319,7 @@
     "min_input_args": 3
   },
   "torch.nn.functional.max_unpool3d": {
-    "Matcher": "UnpoolMatcher",
+    "Matcher": "GenericMatcher",
     "paddle_api": "paddle.nn.functional.max_unpool3d",
     "args_list": [
       "input",

diff --git a/paconvert/api_matcher.py b/paconvert/api_matcher.py
@@ -3878,13 +3878,6 @@ def generate_code(self, kwargs):
         return code
 
 
-class UnpoolMatcher(BaseMatcher):
-    def generate_code(self, kwargs):
-        kwargs["indices"] = "{}.astype('int32')".format(kwargs["indices"])
-
-        return GenericMatcher.generate_code(self, kwargs)
-
-
 class SoftmaxMatcher(BaseMatcher):
     def generate_aux_code(self):
         CODE_TEMPLATE = textwrap.dedent(