Skip to content

Commit 9e25fff

Browse files
committed
2 parents cdcf20e + a2ba9fa commit 9e25fff

2 files changed

Lines changed: 2 additions & 11 deletions

File tree

paconvert/api_mapping.json

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9112,16 +9112,7 @@
91129112
"Matcher": "ChangePrefixMatcher"
91139113
},
91149114
"torch.nn.functional.pixel_unshuffle": {
9115-
"Matcher": "GenericMatcher",
9116-
"paddle_api": "paddle.nn.functional.pixel_unshuffle",
9117-
"min_input_args": 2,
9118-
"args_list": [
9119-
"input",
9120-
"downscale_factor"
9121-
],
9122-
"kwargs_change": {
9123-
"input": "x"
9124-
}
9115+
"Matcher": "ChangePrefixMatcher"
91259116
},
91269117
"torch.nn.functional.poisson_nll_loss": {
91279118
"Matcher": "SizeAverageMatcher",

paconvert/api_matcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ def get_paddle_nodes(self, args, kwargs):
382382
kwargs = self.parse_kwargs(kwargs, allow_none=True)
383383

384384
# temporary delete these unsupport args, which paddle does not support now
385-
for k in ["layout", "generator", "memory_format", "sparse_grad"]:
385+
for k in ["layout", "generator", "memory_format", "sparse_grad", "foreach"]:
386386
if k in kwargs:
387387
kwargs.pop(k)
388388
code = f"{self.get_paddle_api()}({self.args_and_kwargs_to_str(args, kwargs)})"

0 commit comments

Comments
 (0)