density_clustering_cuda_kernels.hpp
1 #pragma once
2 
4 
5 // for pops
6 #define BSIZE_POPS 512
7 
8 // for neighborhood search
9 #define BSIZE_NH 128
10 
11 // for screening
12 #define BSIZE_SCR 256
13 
15 
16 
17 namespace Clustering {
18 namespace Density {
19 namespace CUDA {
20 namespace Kernel {
21 
22  __global__ void
23  population_count(unsigned int offset
24  , float* coords
25  , unsigned int n_rows
26  , unsigned int n_cols
27  , float* radii2
28  , unsigned int n_radii
29  , unsigned int* pops
30  , unsigned int i_from
31  , unsigned int i_to);
32 
33  __global__ void
34  nearest_neighbor_search(unsigned int offset
35  , float* coords
36  , unsigned int n_rows
37  , unsigned int n_cols
38  , float* fe
39  , unsigned int* nh_nhhd_ndx
40  , float* nh_nhhd_dist
41  , unsigned int i_from
42  , unsigned int i_to);
43 
44 
45  __global__ void
46  screening(unsigned int offset
47  , float* sorted_coords
48  , unsigned int n_rows
49  , unsigned int n_cols
50  , float max_dist2
51  , unsigned int* clustering
52  , unsigned int i_from
53  , unsigned int i_to);
54 
55 }}}} // end Clustering::Density::CUDA::Kernel
56 
Clustering
general namespace for clustering package
Definition: coring.cpp:38
Clustering::Density::screening
std::vector< std::size_t > screening(const std::vector< float > &free_energy, const Neighborhood &nh, const float free_energy_threshold, const float *coords, const std::size_t n_rows, const std::size_t n_cols, const std::vector< std::size_t > initial_clusters)
Definition: density_clustering_common.cpp:38
Generated on Wed Jan 16 2019 13:44:40 for clustering by   doxygen 1.8.13