skip api check gen for assign_out_

PaddlePaddle · Dec 5, 2023 · 54c07ff · 54c07ff
1 parent a1bac3d
commit 54c07ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/pir/dialect/op_generator/api_gen.py b/paddle/fluid/pir/dialect/op_generator/api_gen.py
@@ -555,7 +555,7 @@ def _gen_check_data_type(self, op_info, op_name):
 
         if (
             op_name.endswith(('_grad', '_grad_', '_grad_dense', '_grad_sparse'))
-            or op_name in ["print", "hardshrink", "det"]
+            or op_name in ["print", "hardshrink", "det", "assign_out_"]
             or len(mapping_name_to_type) == 0
         ):
             return ""