Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
hust17yixuan committed Aug 28, 2024
1 parent 93127c4 commit b6e76b7
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ void three_nn_forward_npu(int b, int n, int m, const Tensor unknown,

bool is_from_knn = false;
EXEC_NPU_CMD(aclnnKnn, source, target, is_from_knn, dist2);
}

void three_nn_forward_impl(int b, int n, int m, const Tensor unknown,
const Tensor known, Tensor dist2, Tensor idx);

Expand Down

0 comments on commit b6e76b7

Please sign in to comment.