breakgraph when using random.*

Author: DrRyanHuang

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

@@ -70,6 +70,7 @@
     UnsupportedNumPyAPIBreak,
     UnsupportedOperationBreak,
     UnsupportedPaddleAPIBreak,
+    UnsupportedRandomAPIBreak,
 )
 from ..dispatcher import Dispatcher
 from ..guard import (
@@ -250,6 +251,10 @@ def handle_psdb_function(self, /, *args, **kwargs):
         return None
 
     def call_function(self, /, *args, **kwargs) -> VariableBase:
+        if UserDefinedFunctionVariable.__is_random_function(self.value):
+            raise BreakGraphError(
+                UnsupportedRandomAPIBreak(fn_name=self.value.__name__)
+            )
         from ..opcode_inline_executor import OpcodeInlineExecutor
 
         result = self.handle_psdb_function(*args, **kwargs)
@@ -318,6 +323,15 @@ def main_info(self) -> dict[str, Any]:
             "name": self.value.__name__,
         }
 
+    @staticmethod
+    def __is_random_function(value) -> bool:
+        import random
+
+        return value.__qualname__ in [
+            f"{random._inst.__class__.__name__}.{name}"
+            for name in dir(random._inst)
+        ]
+
 
 class UserCodeVariable(FunctionVariable):
     """

python/paddle/jit/sot/utils/exceptions.py

@@ -125,6 +125,25 @@ def __init__(
         )
 
 
+class UnsupportedRandomAPIBreak(UnsupportedOperationBreak):
+    def __init__(
+        self,
+        *,
+        fn_name=None,
+        reason_str=None,
+        file_path="",
+        line_number=-1,
+    ):
+        if reason_str is None:
+            reason_str = f"Random function {fn_name} is not supported."
+
+        super().__init__(
+            reason_str=reason_str,
+            file_path=file_path,
+            line_number=line_number,
+        )
+
+
 class BuiltinFunctionBreak(UnsupportedOperationBreak):
     """Break reason for unsupported built-in function calls.
 

test/sot/test_force_breakgraph_api.py

@@ -0,0 +1,63 @@
+# Copyright (c) 2025 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.
+
+from __future__ import annotations
+
+import random
+import unittest
+
+from test_case_base import TestCaseBase
+
+import paddle
+from paddle.jit.sot import symbolic_translate
+
+
+def fn_randint(x):
+    x = x + 1
+    x = random.randint(0, 100)
+    x = x + 2
+    return x
+
+
+def fn_random(x):
+    x = x + 3
+    x = random.random()
+    x = x + 4
+    return x
+
+
+class TestRandom(TestCaseBase):
+    def test_random_randint(self):
+        x = paddle.to_tensor(2024)
+
+        random.seed(2025)
+        sym_output = symbolic_translate(fn_randint)(x)
+        random.seed(2025)
+        paddle_output = fn_randint(x)
+
+        self.assertEqual(sym_output, paddle_output)
+
+    def test_random_random(self):
+        x = paddle.to_tensor(2025)
+
+        random.seed(2025)
+        sym_output = symbolic_translate(fn_random)(x)
+        random.seed(2025)
+        paddle_output = fn_random(x)
+
+        self.assertEqual(sym_output, paddle_output)
+
+
+if __name__ == "__main__":
+    unittest.main()