From ffb14a147794ef00d8954c97f9a8af09fa8b5228 Mon Sep 17 00:00:00 2001 From: Yue Zhao Date: Mon, 24 Apr 2017 18:48:38 -0400 Subject: [PATCH] add tDot --- parallel/para_gibbs.cu | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/parallel/para_gibbs.cu b/parallel/para_gibbs.cu index c7f08bc..ac50c70 100644 --- a/parallel/para_gibbs.cu +++ b/parallel/para_gibbs.cu @@ -87,7 +87,7 @@ __global__ void seqMetroProcess(int K, int nBlocks, int *y, float *n, curandStat __device__ void sample_theta_seq(float *theta, float *log_theta, int *y, float *n, float a, float b, int K, curandState *state); -__global__ void mergePosterior(int trials, float *dev_a_out,float *dev_b_out); +__global__ void mergePosterior(int trials, float *dev_a_out,float *dev_b_out,int *tDot); int main(int argc, char **argv){ @@ -142,8 +142,8 @@ int main(int argc, char **argv){ //seqMetroProcess<<>>(K,nBlocks,dev_y,dev_n,devStates,dev_theta,dev_log_theta,a,b,trials); seqMetroProcess<<>>(K,nBlocks,dev_y,dev_n,devStates,dev_theta,dev_log_theta,a,b,dev_a_out,dev_b_out,trials); - - mergePosterior<<<1,1>>>(trials,dev_a_out,dev_b_out); + int *tDot; + mergePosterior<<<1,1>>>(trials,dev_a_out,dev_b_out,tDot); /*------ Free Memory -------------------------------------------*/ free(y); @@ -158,7 +158,7 @@ int main(int argc, char **argv){ return EXIT_SUCCESS; } -__global__ void mergePosterior(int trials, float *dev_a_out,float *dev_b_out){ +__global__ void mergePosterior(int trials, float *dev_a_out,float *dev_b_out,int *tDot){ /* printf("\n all blocks finished\n"); for(int j = 0; j < trials ; j++) { printf(" %f ", *dev_a_out);