contiguous support to_static (#61401)

wanghuancoder · web-flow · commit f129cda8b8e9 · 2024-02-01T15:58:23.000+08:00
diff --git a/python/paddle/base/layers/math_op_patch.py b/python/paddle/base/layers/math_op_patch.py
@@ -264,6 +264,30 @@ def place(self):
             "Variable do not have 'place' interface for static graph mode, try not to use it. None will be returned."
         )
 
+    @static_only
+    def contiguous(self):
+        """
+        Variable don't have 'contiguous' interface in static graph mode
+        But this interface can greatly facilitate dy2static.
+        So we give a warnning here and return None.
+        """
+        warnings.warn(
+            "Variable do not have 'contiguous' interface for static graph mode, try not to use it. self will be returned."
+        )
+        return self
+
+    @static_only
+    def is_contiguous(self):
+        """
+        Variable don't have 'is_contiguous' interface in static graph mode
+        But this interface can greatly facilitate dy2static.
+        So we give a warnning here and return None.
+        """
+        warnings.warn(
+            "Variable do not have 'is_contiguous' interface for static graph mode, try not to use it. True will be returned."
+        )
+        return True
+
     def astype(self, dtype):
         """
         **Notes**:
@@ -716,6 +740,8 @@ def to_dense(var):
         ('cpu', cpu),
         ('cuda', cuda),
         ('place', place),
+        ('contiguous', contiguous),
+        ('is_contiguous', is_contiguous),
         ('append', append),
         ('item', _item),
         ('pop', pop),
diff --git a/test/dygraph_to_static/test_contiguous.py b/test/dygraph_to_static/test_contiguous.py
@@ -0,0 +1,47 @@
+# Copyright (c) 2024 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
+#
+#     https://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 unittest
+
+import numpy as np
+from dygraph_to_static_utils import enable_to_static_guard
+
+import paddle
+from paddle.jit.api import to_static
+
+
+@to_static
+def func_test_to_static():
+    x = paddle.arange(12, dtype=paddle.float32).reshape([2, 3, 2])
+    x = x.transpose([1, 0, 2])
+    x = x.contiguous()
+    assert x.is_contiguous()
+    return x
+
+
+class TestContiguous(unittest.TestCase):
+    def test_to_static(self):
+        with enable_to_static_guard(True):
+            static_result = func_test_to_static()
+        with enable_to_static_guard(False):
+            dygraph_result = func_test_to_static()
+        np.testing.assert_allclose(
+            static_result,
+            dygraph_result,
+            err_msg=f'static_result: {static_result} \n dygraph_result: {dygraph_result}',
+        )
+
+
+if __name__ == '__main__':
+    unittest.main()
