[ascend]zq/reimpl index_put (#1009)

reimpl index_put
DeepLink-org · Mar 1, 2024 · 28bbbf4 · 28bbbf4
1 parent 57a3b7a
commit 28bbbf4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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();
 }