Skip to content

Commit 618cf06

Browse files
committed
fix
1 parent f7d0cff commit 618cf06

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

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", "requires_grad", "pin_memory", "device"]:
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)