X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=Gibbs.cpp;fp=Gibbs.cpp;h=f911da53c3e95370a224b95d64250abb3434bb7c;hb=4a5e5138d3fc409e7ade5c14de7689612290f74f;hp=1a084d6b2828731daefe71d8b591d686fa13c19c;hpb=635ca2939cfb1f519f19e9dec072ddd05e9fb450;p=rsem.git

diff --git a/Gibbs.cpp b/Gibbs.cpp
index 1a084d6..f911da5 100644
--- a/Gibbs.cpp
+++ b/Gibbs.cpp
@@ -130,6 +130,7 @@ void load_data(char* reference_name, char* statName, char* imdName) {
 void init() {
 	int quotient, left;
 	char outF[STRLEN];
+	char splitF[STRLEN];
 
 	quotient = NSAMPLES / nThreads;
 	left = NSAMPLES % nThreads;
@@ -156,6 +157,14 @@ void init() {
 		memset(paramsArray[i].pme_theta, 0, sizeof(double) * (M + 1));
 	}
 
+	// output task splitting information
+	sprintf(splitF, "%s.split", imdName);
+	FILE *fo = fopen(splitF, "w");
+	fprintf(fo, "%d", nThreads);
+	for (int i = 0; i < nThreads; i++) fprintf(fo, " %d", paramsArray[i].nsamples);
+	fprintf(fo, "\n");
+	fclose(fo);
+
 	/* set thread attribute to be joinable */
 	pthread_attr_init(&attr);
 	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);