EMAN2
cuda_mpi_kmeans.h
Go to the documentation of this file.
00001 #ifndef cuda_mpi_kmeans_h__
00002 #define cuda_mpi_kmeans_h__ 1
00003 
00004 int cuda_inittest(int numdev);
00005 int cuda_readinit();
00006 int cuda_mpi_init(float* h_IM, float** hd_IM, float** hd_AVE, float** hd_DIST, int size_IM, int size_AVE, int size_DIST, int numdev);
00007 int cuda_mpi_dist(float *h_AVE, float* d_AVE, float* h_DIST, float* d_DIST, float* d_IM, int N, int K, int m);
00008 int cuda_mpi_kmeans(float* h_AVE, float* d_AVE, float* h_DIST, float* d_DIST, float* d_IM, float* h_IM2, float* h_AVE2, unsigned short int* h_ASG, unsigned int* h_NC, int* params);
00009 //int cuda_mpi_kmeans_dist_SSE(float* h_AVE, float* d_AVE, float* h_DIST, float* d_DIST, float* d_IM, float* h_IM2, float* h_AVE2, unsigned short int* h_ASG, unsigned int* h_NC, int* params);
00010 //int cuda_mpi_kmeans_copy_ave_from_device(float* h_AVE, float* d_AVE,int* params);
00011 //int cuda_mpi_kmeans_SSE(float* h_AVE, float* d_AVE, float* h_DIST, float* d_DIST, float* d_IM, float* h_IM2, float* h_AVE2, unsigned short int* h_ASG, unsigned int* h_NC, int* params, int ite, float &ttt);
00012 int cuda_mpi_shutdown(float* d_IM, float* d_AVE, float* d_DIST);
00013 int cuda_mpi_kmeans_SA(float* h_AVE, float* d_AVE, float* h_DIST, float* d_DIST, float* d_IM, float* h_IM2, float* h_AVE2, unsigned short int* h_ASG, unsigned int* h_NC, float T0, int* params);
00014 
00015 #endif // cuda_mpi_kmeans_h__ 1
00016