-
Notifications
You must be signed in to change notification settings - Fork 29
/
Copy pathupdate_param.h
18 lines (12 loc) · 1.14 KB
/
update_param.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#include "cuda.h"
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#ifndef MY_SP_STRUCT
#define MY_SP_STRUCT
#include "my_sp_struct.h"
#endif
__global__ void clear_fields(superpixel_params* sp_params, superpixel_GPU_helper* sp_gpu_helper, const int nsuperpixel, const int nsuperpixel_buffer);
__global__ void sum_by_label(const float* image_gpu_double, const int* seg_gpu, superpixel_params* sp_params, superpixel_GPU_helper* sp_gpu_helper, const int nPixels, const int xdim);
__global__ void calculate_mu_and_sigma(superpixel_params* sp_params, superpixel_GPU_helper* sp_gpu_helper, const int nsuperpixel, const int nsuperpixel_buffer, const int prior_sigma_s, const int prior_count);
__host__ void update_param(const float* image_gpu_double,const int* seg_gpu, superpixel_params* sp_params, superpixel_GPU_helper* sp_gpu_helper, const int nPixels, const int nSps, const int nSps_buffer, const int xdim, const int ydim, const int prior_sigma_s, const int prior_count);
__global__ void reduce0(const float* image_gpu_double, const int* seg_gpu, superpixel_params* sp_params, superpixel_GPU_helper* sp_gpu_helper, const int nPixels, const int xdim);