ACloudViewer  3.9.4
A Modern Library for 3D Data Processing
BallQuery.cuh
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // - CloudViewer: www.cloudViewer.org -
3 // ----------------------------------------------------------------------------
4 // Copyright (c) 2018-2024 www.cloudViewer.org
5 // SPDX-License-Identifier: MIT
6 // ----------------------------------------------------------------------------
7 
8 #pragma once
9 
10 namespace cloudViewer {
11 namespace ml {
12 namespace contrib {
13 
14 __global__ void ball_query_kernel(int b,
15  int n,
16  int m,
17  float radius,
18  int nsample,
19  const float *__restrict__ new_xyz,
20  const float *__restrict__ xyz,
21  int *__restrict__ idx);
22 
23 } // namespace contrib
24 } // namespace ml
25 } // namespace cloudViewer