From 28bbbf47a64a6f9c9620221504a1b36cb712d02c Mon Sep 17 00:00:00 2001
From: ZhangQiu <100055343+NeosZhang@users.noreply.github.com>
Date: Fri, 1 Mar 2024 13:52:08 +0800
Subject: [PATCH] [ascend]zq/reimpl index_put (#1009)

reimpl index_put
---
 impl/ascend_npu/diopi_impl/index_put.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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();
 }