You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
2.1 KiB
53 lines
2.1 KiB
7 years ago
|
#ifndef SERIAL_GPU_UTILS_H /* Include guard */
|
||
|
|
#define SERIAL_GPU_UTILS_H
|
||
|
|
|
||
|
|
#include "meanshift_kernels.h"
|
||
|
|
|
||
|
|
//GPU error check snippet taken from:
|
||
|
|
//https://stackoverflow.com/a/14038590
|
||
|
|
#define gpuErrchk(ans) { gpuAssert((ans), __FILE__, __LINE__); }
|
||
|
|
inline void gpuAssert(cudaError_t code, const char *file, int line, bool abort=true){
|
||
|
|
if (code != cudaSuccess){
|
||
|
|
fprintf(stderr,"GPUassert: %s %s %d\n", cudaGetErrorString(code), file, line);
|
||
|
|
if (abort) exit(code);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Global variables */
|
||
|
|
extern int DEVIATION;
|
||
|
|
extern int NUMBER_OF_POINTS;
|
||
|
|
extern int DIMENSIONS;
|
||
|
|
extern char* POINTS_FILENAME;
|
||
|
|
extern char* LABELS_FILENAME;
|
||
|
|
extern parameters params;
|
||
|
|
extern cudaDeviceProp device_properties;
|
||
|
|
|
||
|
|
void set_GPU();
|
||
|
|
|
||
|
|
//Function meanshift recursively shifts original points according to th
|
||
|
|
//mean-shift algorithm saving the result to shiftedPoints. Struct opt has user
|
||
|
|
//options, h is the desirable deviation.
|
||
|
|
int meanshift(double **original_points, double ***shifted_points, int h
|
||
|
|
, parameters *opt);
|
||
|
|
|
||
|
|
void init_device_memory(double **original_points, double **shifted_points,
|
||
|
|
Matrix *d_original_points, Matrix *d_shifted_points,
|
||
|
|
Matrix *d_kernel_matrix, Matrix *d_denominator, Matrix *d_new_shift);
|
||
|
|
|
||
|
|
void calculate_kernel_matrix(Matrix d_shifted_points, Matrix d_original_points,
|
||
|
|
Matrix d_kernel_matrix, double deviation, double ***kernel_matrix);
|
||
|
|
|
||
|
|
//Function multiply allocates memory in GPU, sends the data and calls the
|
||
|
|
//multiply kernel function.
|
||
|
|
void shift_points(Matrix d_kernel_matrix, Matrix d_original_points, Matrix d_shifted_points,
|
||
|
|
Matrix d_new_shift, Matrix d_denominator, Matrix d_mean_shift_vector, double **kernel_matrix,
|
||
|
|
double **original_points, double ***new_shift, double ***mean_shift_vector);
|
||
|
|
|
||
|
|
void free_device_memory(Matrix d_original_points, Matrix d_kernel_matrix, Matrix d_denominator,
|
||
|
|
Matrix d_new_shift);
|
||
|
|
|
||
|
|
//Function calculate_denominator allocates memory in GPU, sends the data and calls the
|
||
|
|
//denominator kernel function.
|
||
|
|
void calculate_denominator(Matrix d_kernel_matrix, Matrix d_denominator, double **denominator);
|
||
|
|
|
||
|
|
#endif //SERIAL_GPU_UTILS_H
|