Skip to content

Commit

Permalink
Add the support for stack_group_points ops with Ascend on 1.x branch (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
673958639 authored Feb 2, 2024
1 parent 5494299 commit 81f38f4
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions mmcv/ops/csrc/pytorch/npu/stack_group_points_npu.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
#include "pytorch_npu_helper.hpp"

using namespace NPU_NAME_SPACE;
using namespace std;

void stack_group_points_forward_npu(int b, int c, int n, int nsample,
const Tensor features_tensor,
const Tensor features_batch_cnt_tensor,
const Tensor idx_tensor,
const Tensor idx_batch_cnt_tensor,
Tensor out_tensor) {
EXEC_NPU_CMD(aclnnStackGroupPoints, features_tensor,
features_batch_cnt_tensor, idx_tensor, idx_batch_cnt_tensor,
out_tensor);
}

void stack_group_points_forward_impl(int b, int c, int n, int nsample,
const Tensor features_tensor,
const Tensor features_batch_cnt_tensor,
const Tensor idx_tensor,
const Tensor idx_batch_cnt_tensor,
Tensor out_tensor);

REGISTER_NPU_IMPL(stack_group_points_forward_impl,
stack_group_points_forward_npu);

0 comments on commit 81f38f4

Please sign in to comment.