diff --git a/test_tipc/configs/dselect_k/to_static.py b/test_tipc/configs/dselect_k/to_static.py
index 5b589ea03..909ba0d22 100644
--- a/test_tipc/configs/dselect_k/to_static.py
+++ b/test_tipc/configs/dselect_k/to_static.py
@@ -14,6 +14,7 @@
 
 import paddle
 import os
+os.environ['FLAGS_enable_pir_api'] = '0'
 import paddle.nn as nn
 import time
 import logging
diff --git a/test_tipc/configs/esmm/to_static.py b/test_tipc/configs/esmm/to_static.py
index a79fdcf40..9138b0fff 100644
--- a/test_tipc/configs/esmm/to_static.py
+++ b/test_tipc/configs/esmm/to_static.py
@@ -14,6 +14,7 @@
 
 import paddle
 import os
+os.environ['FLAGS_enable_pir_api'] = '0'
 import paddle.nn as nn
 import time
 import logging
diff --git a/test_tipc/configs/mmoe/to_static.py b/test_tipc/configs/mmoe/to_static.py
index 4956eb0f1..2ebd6ea1d 100644
--- a/test_tipc/configs/mmoe/to_static.py
+++ b/test_tipc/configs/mmoe/to_static.py
@@ -14,6 +14,7 @@
 
 import paddle
 import os
+os.environ['FLAGS_enable_pir_api'] = '0'
 import paddle.nn as nn
 import time
 import logging
diff --git a/test_tipc/configs/ple/to_static.py b/test_tipc/configs/ple/to_static.py
index 4956eb0f1..2ebd6ea1d 100644
--- a/test_tipc/configs/ple/to_static.py
+++ b/test_tipc/configs/ple/to_static.py
@@ -14,6 +14,7 @@
 
 import paddle
 import os
+os.environ['FLAGS_enable_pir_api'] = '0'
 import paddle.nn as nn
 import time
 import logging
diff --git a/tools/static_gpubox_trainer.py b/tools/static_gpubox_trainer.py
index b886a7428..a859dee6d 100755
--- a/tools/static_gpubox_trainer.py
+++ b/tools/static_gpubox_trainer.py
@@ -24,6 +24,7 @@
 import paddle.distributed.fleet.base.role_maker as role_maker
 import paddle
 import os
+os.environ['FLAGS_enable_pir_api'] = '0'
 import warnings
 import logging