Skip to content

Commit 2a9aef1

Browse files
committed
fix Tensor.cuda/requires_grad/ctx.saved_tensors
1 parent 9e25fff commit 2a9aef1

2 files changed

Lines changed: 1 addition & 19 deletions

File tree

paconvert/api_mapping.json

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -818,18 +818,7 @@
818818
"paddle_api": "paddle.Tensor.crows"
819819
},
820820
"torch.Tensor.cuda": {
821-
"Matcher": "Device2IntMatcher",
822-
"paddle_api": "paddle.Tensor.cuda",
823-
"min_input_args": 0,
824-
"args_list": [
825-
"device",
826-
"non_blocking",
827-
"memory_format"
828-
],
829-
"kwargs_change": {
830-
"device": "device_id",
831-
"memory_format": ""
832-
}
821+
"Matcher": "ChangePrefixMatcher"
833822
},
834823
"torch.Tensor.cummax": {
835824
"Matcher": "ChangePrefixMatcher"
@@ -3319,9 +3308,6 @@
33193308
"torch.autograd.function.FunctionCtx.save_for_forward": {
33203309
"min_input_args": 0
33213310
},
3322-
"torch.autograd.function.FunctionCtx.saved_tensors": {
3323-
"Matcher": "ChangePrefixMatcher"
3324-
},
33253311
"torch.autograd.function.FunctionCtx.set_materialize_grads": {
33263312
"Matcher": "ChangePrefixMatcher"
33273313
},

paconvert/converter.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,6 @@
2727

2828
from paconvert.transformer.basic_transformer import BasicTransformer
2929
from paconvert.transformer.import_transformer import ImportTransformer
30-
from paconvert.transformer.tensor_requires_grad_transformer import (
31-
TensorRequiresGradTransformer,
32-
)
3330
from paconvert.transformer.custom_op_transformer import (
3431
PreCustomOpTransformer,
3532
CustomOpTransformer,
@@ -378,7 +375,6 @@ def transfer_file(self, old_path, new_path):
378375
def transfer_node(self, root, file):
379376
transformers = [
380377
ImportTransformer, # import ast transformer
381-
TensorRequiresGradTransformer, # attribute requires_grad transformer
382378
BasicTransformer, # most of api transformer
383379
PreCustomOpTransformer, # pre process for C++ custom op
384380
CustomOpTransformer, # C++ custom op transformer

0 commit comments

Comments
 (0)