mirror of https://github.com/jlizier/jidt
27 lines
1.1 KiB
C
27 lines
1.1 KiB
C
#ifndef GPUKNNLIBRARY_H
|
|
#define GPUKNNLIBRARY_H
|
|
|
|
int cudaFindKnn(int* h_bf_indexes, float* h_bf_distances, float* h_pointset,
|
|
float* h_query, int kth, int thelier, int nchunks, int pointdim,
|
|
int signallength, unsigned int useMaxNorm);
|
|
|
|
int cudaFindKnnSetGPU(int* h_bf_indexes, float* h_bf_distances,
|
|
float* h_pointset, float* h_query, int kth, int thelier, int nchunks,
|
|
int pointdim, int signallength, unsigned int useMaxNorm, int deviceid);
|
|
|
|
int cudaFindRSAll(int* h_bf_npointsrange, float* h_pointset,
|
|
float* h_query, float* h_vecradius, int thelier, int nchunks,
|
|
int pointdim, int signallength, unsigned int useMaxNorm);
|
|
|
|
int cudaFindRSAllSetGPU(int* h_bf_npointsrange, float* h_pointset,
|
|
float* h_query, float* h_vecradius, int thelier, int nchunks,
|
|
int pointdim, int signallength, unsigned int useMaxNorm, int deviceid);
|
|
|
|
int findRadiiAlgorithm2(float *radii, const float *data, const int *indexes,
|
|
unsigned int k, unsigned int dim, unsigned int N);
|
|
|
|
int computeSumDigammas(float *sumDiGammas, int *nx, int *ny, unsigned int N);
|
|
|
|
#endif
|
|
|