diff --git a/.gitmodules b/.gitmodules
index 01cd1a390..27d16d0ee 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,6 +1,7 @@
 [submodule "dipu/third_party/DIOPI"]
 	path = dipu/third_party/DIOPI
-	url = https://github.com/DeepLink-org/DIOPI.git
+	url = https://github.com/DeepLink-org/DIOPI.dev.git
+	branch = buildAten_and_wrapper_func
 [submodule "dipu/third_party/kineto"]
 	path = dipu/third_party/kineto
 	url = https://github.com/DeepLink-org/kineto.git
diff --git a/dipu/scripts/ci/ci_one_iter.sh b/dipu/scripts/ci/ci_one_iter.sh
index ae5012f6d..826ffffee 100644
--- a/dipu/scripts/ci/ci_one_iter.sh
+++ b/dipu/scripts/ci/ci_one_iter.sh
@@ -44,7 +44,7 @@ function clone_needed_repo() {
     DIENGINE=dipu_v0.4.8_one_iter_tool
     TRANSFORMERS=dipu_v4.35.2_one_iter_tool
     LIGHTLLM=dipu_one_iter_tool
-    DEEPLINKEXT=ee45ff3015e616c0dd49b374dc8e3b9bfa6e2601
+    DEEPLINKEXT=2a47138de420a0147e8de70685e628d3732135d7
     ALPACALORA=sco_oneiter_finetune
 
     check_and_clone_repository "DI-engine" ${DIENGINE}