[3.0][Dy2St] Support element_size method for value and breakgraph when access unknown tensor attr (#71605) (#71618)

* [Dy2St] Support `element_size` method for value and breakgraph when access unknown tensor attr (#71605)

* fix hard coded branch name

* set $BRANCH to `release/3.0`

* use `github.base_ref`

* Revert "use `github.base_ref`"

This reverts commit 036a01d.

* Revert "set $BRANCH to `release/3.0`"

This reverts commit 8addc14.

* Revert "fix hard coded branch name"

This reverts commit 827d8c9.

Author: SigureMo

paddle/fluid/pybind/pir.cc

@@ -1556,6 +1556,8 @@ void BindValue(py::module *m) {
       .def("apply", &apply)
       .def("is_same", &Value::operator==)
       .def("hash", [](Value self) { return std::hash<pir::Value>{}(self); })
+      .def("element_size",
+           [](Value self) { return phi::SizeOf(pir::GetValueDtype(self)); })
       .def("_rename", &name_analysis::RenameValue)
       .def("_has_only_one_name",
            [](Value self) -> bool {

python/paddle/jit/sot/opcode_translator/executor/opcode_executor.py

@@ -899,6 +899,7 @@ def load_method(self, method_name):
             if CALL_METHOD_LAYOUT_NULL_AFTER_VALUE:
                 self.stack.push(NullVariable())
 
+    @call_break_graph_decorator(push_n=2)
     def LOAD_METHOD(self, instr: Instruction):
         method_name = self._code.co_names[instr.arg]
         self.load_method(method_name)

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

@@ -48,7 +48,10 @@
     printable,
 )
 from ....utils.envs import ENV_SOT_BREAK_GRAPH_ON_GET_SYMBOLIC_VALUE
-from ....utils.exceptions import HasNoAttributeError, InnerError
+from ....utils.exceptions import (
+    InnerError,
+    UnsupportedPaddleAPIBreak,
+)
 from ..dispatch_functions import tensor_numel
 from ..guard import (
     FasterStringifiedExpression,
@@ -704,7 +707,9 @@ def getattr(self, name: str, default=None):
             )
             return fn_var.bind(self, name)
         else:
-            raise HasNoAttributeError(f"Unknown Tensor attribute: {name}")
+            raise BreakGraphError(
+                UnsupportedPaddleAPIBreak(fn_name=f"Tensor.{name}")
+            )
 
     def setattr(self, key, val):
         # support tensor variable store attr, like:

test/dygraph_to_static/test_tensor_attr_consistency.py

@@ -37,7 +37,6 @@
         'data',
         'data_ptr',
         'detach_',
-        'element_size',
         'fill_',
         'fill_diagonal_',
         'fill_diagonal_tensor',

test/sot/test_18_tensor_method.py

@@ -57,6 +57,7 @@ def tensor_method_property_without_breakgraph(
         + a.ndim
         + a.dim()
         + a.rank(),
+        a.element_size(),
     )
 
 
@@ -67,6 +68,7 @@ def tensor_method_property_with_breakgraph(a: paddle.Tensor, b: paddle.Tensor):
         a.tolist(),
         str(a.place),
         a.clear_gradient(),
+        a.is_dense(),
     )
 
 

test/sot/test_sot_exception.py

@@ -22,7 +22,7 @@
 
 
 def case1(x):
-    return n  # noqa: F821
+    return undefined_var  # noqa: F821
 
 
 def case2(x):
@@ -31,15 +31,15 @@ def case2(x):
 
 
 def case3(x):
-    y = x.undefined_attr
+    y = undefined_var  # noqa: F821
     return y
 
 
 def case4_inner(x):
     y = x * 2
     print()
     y = y + 1
-    return y.undefined_attr
+    return undefined_var  # noqa: F821
 
 
 def case4(x):