diff --git a/impl/ascend_npu/diopi_impl/index_put.cpp b/impl/ascend_npu/diopi_impl/index_put.cpp
index 092a72b56..a09f26118 100644
--- a/impl/ascend_npu/diopi_impl/index_put.cpp
+++ b/impl/ascend_npu/diopi_impl/index_put.cpp
@@ -6,6 +6,7 @@
 
 #include "helper.hpp"
 #include "op_plugin/AclOpsInterface.h"
+#include "op_plugin/OpApiInterface.h"
 
 namespace OP_IMPL_NS {
 
@@ -19,7 +20,7 @@ diopiError_t diopiIndexPut(diopiContextHandle_t ctx, diopiTensorHandle_t out, di
     }
 
     outAt.copy_(inputAt);
-    at::index_put_(outAt, indicesAtList, valuesAt, accumulate);
+    op_api::index_put_(outAt, indicesAtList, valuesAt, accumulate);
     END_CALL_ACL_OP();
 }
 
@@ -32,7 +33,7 @@ diopiError_t diopiIndexPutInp(diopiContextHandle_t ctx, diopiTensorHandle_t inpu
         indicesAtList.emplace_back(impl::aten::buildATen(indices[i]));
     }
 
-    at::index_put_(inputAt, indicesAtList, valuesAt, accumulate);
+    op_api::index_put_(inputAt, indicesAtList, valuesAt, accumulate);
     END_CALL_ACL_OP();
 }