Skip to content

Commit 9437361

Browse files
Charles-hitco63oc
authored andcommitted
add more recompute ops (PaddlePaddle#63272)
1 parent 8d92ed6 commit 9437361

File tree

1 file changed

+40
-0
lines changed

1 file changed

+40
-0
lines changed

python/paddle/decomposition/recompute.py

+40
Original file line numberDiff line numberDiff line change
@@ -76,11 +76,51 @@
7676
"pd_op.where",
7777
"pd_op.prod",
7878
"pd_op.log",
79+
"pd_op.log1p",
80+
"pd_op.logit",
7981
"pd_op.max",
8082
"pd_op.expand_as",
8183
"pd_op.split",
8284
"pd_op.arange",
8385
"pd_op.put_along_axis",
86+
"pd_op.tanh",
87+
"pd_op.atan",
88+
"pd_op.atanh",
89+
"pd_op.sinh",
90+
"pd_op.asin",
91+
"pd_op.asinh",
92+
"pd_op.cosh",
93+
"pd_op.acos",
94+
"pd_op.acosh",
95+
"pd_op.abs",
96+
"pd_op.sign",
97+
"pd_op.expm1",
98+
"pd_op.erf",
99+
"pd_op.erfinv",
100+
"pd_op.ceil",
101+
"pd_op.floor",
102+
"pd_op.frac",
103+
"pd_op.round",
104+
"pd_op.trunc",
105+
"pd_op.equal",
106+
"pd_op.angle",
107+
"pd_op.as_complex",
108+
"pd_op.as_real",
109+
"pd_op.complex",
110+
"pd_op.real",
111+
"pd_op.imag",
112+
"pd_op.conj",
113+
"pd_op.not_equal",
114+
"pd_op.greater_equal",
115+
"pd_op.greater_than",
116+
"pd_op.less_equal",
117+
"pd_op.less_than",
118+
"pd_op.bitwise_and",
119+
"pd_op.bitwise_not",
120+
"pd_op.bitwise_or",
121+
"pd_op.bitwise_xor",
122+
"pd_op.isinf",
123+
"pd_op.isnan",
84124
]
85125

86126
VIEW_OPS: List[str] = []

0 commit comments

Comments
 (0)