diff --git a/paddle/phi/core/flags.cc b/paddle/phi/core/flags.cc
index f11e09cf890e892745b2bd903e28112e85d635f9..ee3caeea36766c5f1c3f29c3173c7a6906e321ed 100644
--- a/paddle/phi/core/flags.cc
+++ b/paddle/phi/core/flags.cc
@@ -1040,7 +1040,6 @@ PADDLE_DEFINE_EXPORTED_string(jit_engine_type,
                               "Predictor",
                               "Choose default funciton type in JitLayer.");
 
-#ifdef PADDLE_WITH_CUSTOM_DEVICE
 /**
  * Custom Device NPU related FLAG
  * Name: FLAGS_npu_storage_format
@@ -1050,7 +1049,6 @@ PADDLE_DEFINE_EXPORTED_string(jit_engine_type,
  * Note: Enable NPU Storage Format for Ascend910 performance improvement.
  */
 PADDLE_DEFINE_EXPORTED_bool(npu_storage_format, false, "");
-#endif
 
 #ifdef PADDLE_WITH_CUDNN_FRONTEND
 /**
diff --git a/python/paddle/fluid/dygraph/varbase_patch_methods.py b/python/paddle/fluid/dygraph/varbase_patch_methods.py
index f52ba97066c1b1390c61a0411b0a2a1cbf1d61a6..9ddebbab7670eb4b353bceef6d4eec0af795414f 100644
--- a/python/paddle/fluid/dygraph/varbase_patch_methods.py
+++ b/python/paddle/fluid/dygraph/varbase_patch_methods.py
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import inspect
 import numpy as np
 import warnings
@@ -42,6 +41,7 @@ import paddle.profiler as profiler
 from paddle.profiler.utils import in_profiler_mode
 from paddle import _C_ops, _legacy_C_ops
 from paddle.device import get_all_custom_device_type
+from paddle.fluid.framework import _global_flags
 
 _grad_scalar = None
 
@@ -381,8 +381,7 @@ def monkey_patch_varbase():
             new_ivar = self._grad_ivar()
             # TODO(qili93): temporary for ascned npu performance to be removed along with npu_identity op
             if (
-                os.environ.get('FLAGS_npu_storage_format', None)
-                in [1, '1', True, 'True', 'true']
+                _global_flags()['FLAGS_npu_storage_format']
                 and 'npu' in get_all_custom_device_type()
             ):
                 new_ivar = paddle.incubate._npu_identity(x=new_ivar, format=-1)
diff --git a/python/paddle/nn/functional/conv.py b/python/paddle/nn/functional/conv.py
index c479cabb4fb8f95136486ad11309e09e5d00e6c0..d6f4ee12eea7f2dab4beb009b87d5844f8c4d86b 100644
--- a/python/paddle/nn/functional/conv.py
+++ b/python/paddle/nn/functional/conv.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
-
 from paddle import _C_ops, _legacy_C_ops, get_flags, in_dynamic_mode
 from paddle.device import (
     get_all_custom_device_type,
@@ -152,8 +150,7 @@ def _conv_nd(
             bias = bias.reshape(new_shape)
             # TODO(qili93): temporary for ascned npu performance to be removed along with npu_identity op
             if (
-                os.environ.get('FLAGS_npu_storage_format', None)
-                in [1, '1', True, 'True', 'true']
+                _global_flags()['FLAGS_npu_storage_format']
                 and 'npu' in get_all_custom_device_type()
             ):
                 with no_grad():
@@ -753,8 +750,7 @@ def conv2d(
                     )
                 # TODO(qili93): temporary for ascned npu performance to be removed along with npu_identity op
                 if (
-                    os.environ.get('FLAGS_npu_storage_format', None)
-                    in [1, '1', True, 'True', 'true']
+                    _global_flags()['FLAGS_npu_storage_format']
                     and 'npu' in get_all_custom_device_type()
                 ):
                     with no_grad():
diff --git a/python/paddle/nn/layer/norm.py b/python/paddle/nn/layer/norm.py
index f446970ee0ef612c75bf923453a6343219800b05..e2842e1944d92214966c5c095992b1fdc75220cc 100644
--- a/python/paddle/nn/layer/norm.py
+++ b/python/paddle/nn/layer/norm.py
@@ -28,7 +28,6 @@
 # TODO: define normalization api
 
 import numbers
-import os
 import warnings
 
 import numpy as np
@@ -688,8 +687,7 @@ class _BatchNormBase(Layer):
 
         # TODO(qili93): temporary for ascned npu performance to be removed along with npu_identity op
         if (
-            os.environ.get('FLAGS_npu_storage_format', None)
-            in [1, '1', True, 'True', 'true']
+            _global_flags()['FLAGS_npu_storage_format']
             and 'npu' in get_all_custom_device_type()
         ):
             with no_grad():