Fix endswith

PaddlePaddle · Sep 11, 2023 · 927aa35 · 927aa35
1 parent 2d639b8
commit 927aa35
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py b/paddle/fluid/pir/dialect/op_generator/ops_api_gen.py
@@ -97,9 +97,7 @@ def __init__(self) -> None:
     def _need_skip(self, op_info, op_name):
         return (
             super()._need_skip(op_info, op_name)
-            or op_name.endswith('_grad')
-            or op_name.endswith('_grad_')
-            or op_name.endswith('xpu')
+            or op_name.endswith(('_grad', '_grad_', 'xpu'))
             or op_name in NO_NEED_GEN_STATIC_ONLY_APIS
         )