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
00010
00011
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