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