cheshirekow  v0.1.0
so3.cu.hpp File Reference

Go to the source code of this file.

Classes

struct  mpblocks::cudaNN::kernels::so3_distance_fn< Scalar, Pseudo >
 
struct  mpblocks::cudaNN::kernels::so3_distance_fn< Scalar, true >
 

Namespaces

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

Enumerations

enum  mpblocks::cudaNN::kernels::Constraint { mpblocks::cudaNN::kernels::OFF = 0, mpblocks::cudaNN::kernels::MIN = 1, mpblocks::cudaNN::kernels::MAX = 2 }
 

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)
 
template<typename Scalar >
__device__ Scalar mpblocks::cudaNN::kernels::so3_distance (const linalg::Matrix< Scalar, 4, 1 > &q0, const linalg::Matrix< Scalar, 4, 1 > &q1)
 
template<typename Scalar >
__device__ Scalar mpblocks::cudaNN::kernels::so3_pseudo_distance (const linalg::Matrix< Scalar, 4, 1 > &q0, const linalg::Matrix< Scalar, 4, 1 > &q1)