Skip to content

Commit 28bbbf4

Browse files
authored
[ascend]zq/reimpl index_put (#1009)
reimpl index_put
1 parent 57a3b7a commit 28bbbf4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

impl/ascend_npu/diopi_impl/index_put.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
#include "helper.hpp"
88
#include "op_plugin/AclOpsInterface.h"
9+
#include "op_plugin/OpApiInterface.h"
910

1011
namespace OP_IMPL_NS {
1112

@@ -19,7 +20,7 @@ diopiError_t diopiIndexPut(diopiContextHandle_t ctx, diopiTensorHandle_t out, di
1920
}
2021

2122
outAt.copy_(inputAt);
22-
at::index_put_(outAt, indicesAtList, valuesAt, accumulate);
23+
op_api::index_put_(outAt, indicesAtList, valuesAt, accumulate);
2324
END_CALL_ACL_OP();
2425
}
2526

@@ -32,7 +33,7 @@ diopiError_t diopiIndexPutInp(diopiContextHandle_t ctx, diopiTensorHandle_t inpu
3233
indicesAtList.emplace_back(impl::aten::buildATen(indices[i]));
3334
}
3435

35-
at::index_put_(inputAt, indicesAtList, valuesAt, accumulate);
36+
op_api::index_put_(inputAt, indicesAtList, valuesAt, accumulate);
3637
END_CALL_ACL_OP();
3738
}
3839

0 commit comments

Comments
 (0)