mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-27 16:00:07 +08:00
erikwijmans
24 lines
605 B
C
24 lines
605 B
C
#ifndef _SAMPLING_GPU_H
|
|
#define _SAMPLING_GPU_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
void gather_points_kernel_wrapper(int b, int c, int n, int npoints,
|
|
const float *points, const int *idx,
|
|
float *out, cudaStream_t stream);
|
|
|
|
void gather_points_grad_kernel_wrapper(int b, int c, int n, int npoints,
|
|
const float *grad_out, const int *idx,
|
|
float *grad_points, cudaStream_t stream);
|
|
|
|
void furthest_point_sampling_kernel_wrapper(int b, int n, int m,
|
|
const float *dataset, float *temp,
|
|
int *idxs, cudaStream_t stream);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif
|