From 9429ec48258d8da5cac176737860c086b6c6b6eb Mon Sep 17 00:00:00 2001
From: wanghuancoder <wanghuan29@baidu.com>
Date: Wed, 2 Aug 2023 10:40:49 +0800
Subject: [PATCH] fix _no_check_dy2st_diff path (#55864)

---
 python/paddle/fluid/framework.py    | 1 +
 python/paddle/framework/__init__.py | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py
index 5ef1ffeb4dc..efb038669e8 100644
--- a/python/paddle/fluid/framework.py
+++ b/python/paddle/fluid/framework.py
@@ -62,6 +62,7 @@ __all__ = [
     'device_guard',
     'set_flags',
     'get_flags',
+    '_stride_in_no_check_dy2st_diff',
 ]
 
 EMPTY_VAR_NAME = core.kEmptyVarName()
diff --git a/python/paddle/framework/__init__.py b/python/paddle/framework/__init__.py
index 16f12fd41ec..778e2b55cd7 100755
--- a/python/paddle/framework/__init__.py
+++ b/python/paddle/framework/__init__.py
@@ -79,5 +79,8 @@ from ..fluid.framework import _set_expected_place  # noqa: F401
 from ..fluid.framework import Block, Program  # noqa: F401
 from ..fluid.framework import IrGraph  # noqa: F401
 from ..fluid.framework import deprecate_stat_dict
+from ..fluid.framework import (
+    _stride_in_no_check_dy2st_diff as _no_check_dy2st_diff,
+)  # noqa: F401
 
 __all__ = []
-- 
GitLab