cheshirekow  v0.1.0
so3.cu.h File Reference
#include <cuda.h>
#include <cuda_runtime.h>
#include <mpblocks/cudaNN/QueryPoint.h>

Go to the source code of this file.

Namespaces

 mpblocks
 
 mpblocks::cudaNN
 
 mpblocks::cudaNN::kernels
 

Functions

template<bool Psuedo, typename Scalar , unsigned int NDim>
__global__ void mpblocks::cudaNN::kernels::so3_distance (QueryPoint< Scalar, NDim > query, Scalar *g_in, unsigned int pitchIn, Scalar *g_out, unsigned int pitchOut, unsigned int n)
 
template<bool Pseudo, typename Scalar , unsigned int NDim>
__global__ void mpblocks::cudaNN::kernels::so3_distance (RectangleQuery< Scalar, NDim > query, Scalar *g_out)