[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 are 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

@@ -40,3 +40,4 @@
 ./test_specialization.py
 ./test_str_format.py
 ./test_tensor_dtype_in_guard.py
+./test_builtin_bool.py

test/sot/test_builtin_bool.py

@@ -0,0 +1,117 @@
+# 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)
+
+
+def call_bool_in_cond(obj):
+    if obj:
+        return True
+    else:
+        return False
+
+
+def call_bool_by_bool(obj):
+    return bool(obj)
+
+
+def call_bool_by_operator_truth(obj):
+    return operator.truth(obj)
+
+
+class TestBuiltinBool(TestCaseBase):
+    def test_object_disallow_breakgraph(self):  # disallow breakgraph
+        call_bool_in_cond_no_breakgraph = check_no_breakgraph(call_bool_in_cond)
+        call_bool_by_bool_no_breakgraph = check_no_breakgraph(call_bool_by_bool)
+        call_bool_by_operator_truth_no_breakgraph = check_no_breakgraph(
+            call_bool_by_operator_truth
+        )
+
+        obj = TestObject()
+        self.assert_results(call_bool_in_cond_no_breakgraph, obj)
+        self.assert_results(call_bool_by_bool_no_breakgraph, obj)
+        self.assert_results(call_bool_by_operator_truth_no_breakgraph, obj)
+
+        obj = TestObjectWithBool()
+        self.assert_results(call_bool_in_cond_no_breakgraph, obj)
+        self.assert_results(call_bool_by_bool_no_breakgraph, obj)
+        self.assert_results(call_bool_by_operator_truth_no_breakgraph, obj)
+
+        obj = TestObjectWithBoolAndLen([1, 2, 3])
+        self.assert_results(call_bool_in_cond_no_breakgraph, obj)
+        self.assert_results(call_bool_by_bool_no_breakgraph, obj)
+        self.assert_results(call_bool_by_operator_truth_no_breakgraph, obj)
+
+        obj = TestObjectWithBoolAndLen([])
+        self.assert_results(call_bool_in_cond_no_breakgraph, obj)
+        self.assert_results(call_bool_by_bool_no_breakgraph, obj)
+        self.assert_results(call_bool_by_operator_truth_no_breakgraph, obj)
+
+        layer = paddle.nn.Linear(10, 1)
+        self.assert_results(call_bool_in_cond_no_breakgraph, layer)
+        self.assert_results(call_bool_by_bool_no_breakgraph, layer)
+        self.assert_results(call_bool_by_operator_truth_no_breakgraph, layer)
+
+    def test_object_allow_breakgraph(self):  # allow breakgraph
+        obj = TestObjectWithLen([1, 2, 3])
+        with strict_mode_guard(False):
+            self.assert_results(call_bool_in_cond, obj)
+
+        self.assert_results(call_bool_by_bool, obj)
+        self.assert_results(call_bool_by_operator_truth, obj)
+
+        obj = TestObjectWithLen([])
+        with strict_mode_guard(False):
+            self.assert_results(call_bool_in_cond, obj)
+
+        self.assert_results(call_bool_by_bool, obj)
+        self.assert_results(call_bool_by_operator_truth, obj)
+
+
+if __name__ == "__main__":
+    unittest.main()