[SOT] support inline call bool magic function

python/paddle/jit/sot/opcode_translator/executor/variables/callable.py

@@ -646,6 +646,16 @@ def call_function(self, /, *args, **kwargs):
                 )
                 assert isinstance(fn_var, VariableBase)
                 return fn_var(*args)
+            # If __bool__ and __len__ method is absent, inline bool calls return True.
+            # See https://github.com/python/cpython/blob/3.11/Objects/typeobject.c#L7463
+            elif magic_method.name == "__bool__" and not hasattr(
+                arg_type, "__len__"
+            ):
+                return VariableFactory.from_value(
+                    True,
+                    self.graph,
+                    DummyTracker([self] + list(args) + list(kwargs.values())),
+                )
 
         # Break graph if neither of the above conditions is met
         arg_types = ", ".join([type(arg).__name__ for arg in args])

test/sot/skip_files_py312

@@ -1,19 +1,14 @@
 ./test_01_basic.py
-./test_02_store_inplace.py
 ./test_03_tuple.py
 ./test_04_list.py
 ./test_05_dict.py
-./test_06_call_function.py
 ./test_07_unpack.py
-./test_08_rot.py
 ./test_09_f_string.py
 ./test_10_build_unpack.py
 ./test_11_jumps.py
 ./test_12_for_loop.py
-./test_13_make_function.py
 ./test_14_operators.py
 ./test_15_slice.py
-./test_16_paddle_api.py
 ./test_17_paddle_layer.py
 ./test_18_tensor_method.py
 ./test_19_closure.py
@@ -26,26 +21,14 @@
 ./test_builtin_map.py
 ./test_builtin_range.py
 ./test_builtin_zip.py
-./test_call_ast.py
-./test_call_object.py
-./test_case_base.py
 ./test_constant_graph.py
-./test_delete_fast.py
-./test_dtype.py
 ./test_dup_top.py
 ./test_enumerate.py
-./test_execution_base.py
-./test_guard_outputs.py
 ./test_guard_user_defined_fn.py
 ./test_inplace_api.py
-./test_instruction_translator_cache.py
 ./test_min_graph_size.py
-./test_model_switch_training.py
-./test_multiple_args.py
-./test_numpy.py
 ./test_numpy_var_if.py
 ./test_output_restoration.py
-./test_segment_linear.py
 ./test_side_effects.py
 ./test_simulate_initialize.py
 ./test_sir_rollback.py
@@ -57,6 +40,4 @@
 ./test_specialization.py
 ./test_str_format.py
 ./test_tensor_dtype_in_guard.py
-./test_tensor_slice.py
-./test_trace_list_arg.py
-./test_unsupport_function.py
+./test_builtin_bool.py

test/sot/test_builtin_bool.py

@@ -0,0 +1,148 @@
+# Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import operator
+import unittest
+
+from test_case_base import TestCaseBase
+
+import paddle
+from paddle.jit.sot.psdb import check_no_breakgraph
+from paddle.jit.sot.utils import strict_mode_guard
+
+
+class TestObject:
+    pass
+
+
+class TestObjectWithBool:
+    def __bool__(self):
+        return False
+
+
+class TestObjectWithLen:
+    def __init__(self, list):
+        self.list = list
+
+    def __len__(self):
+        return len(self.list)
+
+
+class TestObjectWithBoolAndLen:
+    def __init__(self, list):
+        self.list = list
+
+    def __bool__(self):
+        return False
+
+    def __len__(self):
+        return len(self.list)
+
+
+@check_no_breakgraph
+def object_bool(obj):
+    if obj:
+        return True
+    else:
+        return False
+
+
+@strict_mode_guard(False)
+def object_bool_allow_breakgraph(obj):
+    if obj:
+        return True
+    else:
+        return False
+
+
+@check_no_breakgraph
+def test_bool(obj):
+    return bool(obj)
+
+
+@check_no_breakgraph
+def test_operator_truth(obj):
+    return operator.truth(obj)
+
+
+def test_bool_allow_breakgraph(obj):
+    return bool(obj)
+
+
+def test_operator_truth_allow_breakgraph(obj):
+    return operator.truth(obj)
+
+
+class TestBuiltinBool(TestCaseBase):
+    def test_object(self):
+        object = TestObject()
+        self.assert_results(object_bool, object)
+        self.assert_results(object_bool, bool(object))
+        self.assert_results(object_bool, operator.truth(object))
+        self.assert_results(test_bool, object)
+        self.assert_results(test_operator_truth, object)
+
+    def test_object_with_bool(self):
+        object = TestObjectWithBool()
+        self.assert_results(object_bool, object)
+        self.assert_results(object_bool, bool(object))
+        self.assert_results(object_bool, operator.truth(object))
+        self.assert_results(test_bool, object)
+        self.assert_results(test_operator_truth, object)
+
+    def test_object_with_len(self):
+        object = TestObjectWithLen([1, 2, 3])
+        self.assert_results(object_bool_allow_breakgraph, object)
+        self.assert_results(object_bool_allow_breakgraph, bool(object))
+        self.assert_results(
+            object_bool_allow_breakgraph, operator.truth(object)
+        )
+        self.assert_results(test_bool_allow_breakgraph, object)
+        self.assert_results(test_operator_truth_allow_breakgraph, object)
+
+        object = TestObjectWithLen([])
+        self.assert_results(object_bool_allow_breakgraph, object)
+        self.assert_results(object_bool_allow_breakgraph, bool(object))
+        self.assert_results(
+            object_bool_allow_breakgraph, operator.truth(object)
+        )
+        self.assert_results(test_bool_allow_breakgraph, object)
+        self.assert_results(test_operator_truth_allow_breakgraph, object)
+
+    def test_object_with_bool_and_len(self):
+        object = TestObjectWithBoolAndLen([1, 2, 3])
+        self.assert_results(object_bool, object)
+        self.assert_results(object_bool, bool(object))
+        self.assert_results(object_bool, operator.truth(object))
+        self.assert_results(test_bool, object)
+        self.assert_results(test_operator_truth, object)
+
+        object = TestObjectWithBoolAndLen([])
+        self.assert_results(object_bool, object)
+        self.assert_results(object_bool, bool(object))
+        self.assert_results(object_bool, operator.truth(object))
+        self.assert_results(test_bool, object)
+        self.assert_results(test_operator_truth, object)
+
+    def test_layer(self):
+        layer = paddle.nn.Linear(10, 1)
+        self.assert_results(object_bool, layer)
+        self.assert_results(object_bool, bool(layer))
+        self.assert_results(object_bool, operator.truth(layer))
+        self.assert_results(test_bool, layer)
+        self.assert_results(test_operator_truth, layer)
+
+
+if __name__ == "__main__":
+    unittest.main()