so3.cu.h

Go to the documentation of this file.

/*
 *  Copyright (C) 2012 Josh Bialkowski (jbialk@mit.edu)
 *
 *  This file is part of mpblocks.
 *
 *  mpblocks is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  mpblocks is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with mpblocks.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef MPBLOCKS_CUDANN_KERNELS_SO3_CU_H_
#define MPBLOCKS_CUDANN_KERNELS_SO3_CU_H_

#include <cuda.h>
#include <cuda_runtime.h>
#include <mpblocks/cudaNN/QueryPoint.h>

namespace  mpblocks {
namespace    cudaNN {
namespace   kernels {


template< bool Psuedo, typename Scalar, unsigned int NDim >
__global__  void 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 so3_distance(
       RectangleQuery<Scalar,NDim>
                    query,      
       Scalar*      g_out
       );




} // kernels
} // cudaNN
} // mpblocks




#endif // KERNELS_H_