new pipelines for updater and broadcaster

source/train/XWorker.cpp

@@ -142,28 +142,38 @@ void XWorker::DequeueFinished(XList* args)
     worker->DequeueFinishedJob();
 }
 
-/* add a job of enqueuing a counting a finished job */
-void XWorker::AddJobEnqueueFinished()
+/* 
+add a job of enqueuing a counting a finished job
+>> jobQueue - the queue where we push the job
+*/
+void XWorker::AddJobEnqueueFinished(XQueue* jobQueue)
 {
     XList args;
     args.Add(this);
 
+    XQueue& queueRun = jobQueue != NULL ? *jobQueue : queue;
+
     if (isInstantRun)
         XWorker::EnqueueFinished(&args);
     else
-        queue.EnqueueJob((void*)(char*)XWorker::EnqueueFinished, &args);
+        queueRun.EnqueueJob((void*)(char*)XWorker::EnqueueFinished, &args);
 }
 
-/* add a job of dequeuing a counting a finished job */
-void XWorker::AddJobDequeueFinished()
+/* 
+add a job of dequeuing a counting a finished job 
+>> jobQueue - the queue where we push the job
+*/
+void XWorker::AddJobDequeueFinished(XQueue* jobQueue)
 {
     XList args;
     args.Add(this);
 
+    XQueue& queueRun = jobQueue != NULL ? *jobQueue : queue;
+
     if (isInstantRun)
         XWorker::DequeueFinished(&args);
     else
-        queue.EnqueueJob((void*)(char*)XWorker::DequeueFinished, &args);
+        queueRun.EnqueueJob((void*)(char*)XWorker::DequeueFinished, &args);
 
 }
 

source/train/XWorker.h

@@ -122,10 +122,10 @@ public:
     void DequeueFinished(XList* args);
 
     /* add a job of enqueuing a counting a finished job */
-    void AddJobEnqueueFinished();
+    void AddJobEnqueueFinished(XQueue * jobQueue = NULL);
 
     /* add a job of dequeuing a counting a finished job */
-    void AddJobDequeueFinished();
+    void AddJobDequeueFinished(XQueue* jobQueue = NULL);
 };
 
 }

source/train/XWorkerBroadcast.cpp

@@ -51,14 +51,11 @@ void XWorkerBroadcast::SetBroadcastMode(DATA_BROADCAST_TYPE myMode)
 
 /* 
 broadcast data for a parameter 
->> jobQueue - the queue where we push jobs
 >> source - the data (as a model) that we want to broadcast
 >> targetList - the target places that we recieve the data
 >> pid - the parameter index
 */
-void XWorkerBroadcast::BroadcastDataSingle(XQueue * jobQueue, 
-                                           XModel * source, XList * targetList, 
-                                           int pid)
+void XWorkerBroadcast::BroadcastDataSingle(XModel * source, XList * targetList, int pid)
 {
     CheckNTErrors(source->params[pid].flag == PARAM_STATE_UPDATED,
                   "The parameter is not ready for broadcasting");
@@ -83,7 +80,6 @@ void XWorkerBroadcast::BroadcastSingle(XList * args)
     int paramCount = 0;
 
     XWorkerBroadcast * broadcaster = (XWorkerBroadcast*)args->GetItem(paramCount++);
-    XQueue * jobQueue = (XQueue*)args->GetItem(paramCount++);
     XModel * source = (XModel*)args->GetItem(paramCount++);
 
     /* target models */
@@ -97,7 +93,7 @@ void XWorkerBroadcast::BroadcastSingle(XList * args)
     /* parameter index */
     int p = args->GetInt(paramCount++);
 
-    broadcaster->BroadcastDataSingle(jobQueue, source, &target, p);
+    broadcaster->BroadcastDataSingle(source, &target, p);
 }
 
 /* 
@@ -117,7 +113,7 @@ void XWorkerBroadcast::BroadcastP2P(XTensor * source, XTensor * target)
 
 /* 
 add a new job of broadcasting data (for a parameter)
->> jobQueue - the queue that we push jobs here
+>> jobQueue - the queue where we push jobs
 >> source - the data that we want to broadcast
 >> targetList - the target places that we recieve the data
 >> pid - the parameter index
@@ -130,16 +126,17 @@ bool XWorkerBroadcast::AddJobBroadcastSingle(XQueue * jobQueue, XModel * source,
 
     XList args;
     args.Add(this);
-    args.Add(jobQueue);
     args.Add(source);
     args.AddInt(targetList->count);
     args.AddList(targetList);
     args.AddInt(pid);
 
+    XQueue& queueRun = jobQueue != NULL ? *jobQueue : queue;
+
     if (isInstantRun)
         XWorkerBroadcast::BroadcastSingle(&args);
     else
-        queue.EnqueueJob((void*)(char*)XWorkerBroadcast::BroadcastSingle, &args);
+        queueRun.EnqueueJob((void*)(char*)XWorkerBroadcast::BroadcastSingle, &args);
 
     return true;
 }

source/train/XWorkerBroadcast.h

@@ -61,7 +61,7 @@ public:
     void SetBroadcastMode(DATA_BROADCAST_TYPE myMode);
 
     /* broadcast data for a parameter */
-    void BroadcastDataSingle(XQueue * jobQueue, XModel * source, XList * targetList, int pid);
+    void BroadcastDataSingle(XModel * source, XList * targetList, int pid);
 
     /* wrapper of BroadcastDataSingle */
     static

source/train/XWorkerCollect.cpp

@@ -129,10 +129,15 @@ void XWorkerCollect::UpdateDataAll(XList * jobQueues, XList * memberActive, XLis
                         if (finishedCount[j] == memberActive->count) {
                             paramServer.flag = PARAM_STATE_COLLECTED;
                             if (updater != NULL) {
-                                updater->AddJobUpdate((XQueue*)jobQueues->GetItem(j), 
-                                                      server, memberAll, j, 
-                                                      optimizer, broadcaster);
-                                updater->AddJobEnqueueFinished();
+                                XQueue* jobQueue = (XQueue*)jobQueues->GetItem(j);
+
+                                /* update the parameters */
+                                updater->AddJobUpdate(jobQueue, server, j, optimizer);
+                                updater->AddJobEnqueueFinished(jobQueue);
+
+                                /* broadcast the new parameter to other models*/
+                                broadcaster->AddJobBroadcastSingle(jobQueue, server, memberAll, j);
+                                broadcaster->AddJobEnqueueFinished(jobQueue);
                             }
                         }
                         else if (finishedCount[j] > memberActive->count) {

source/train/XWorkerUpdate.cpp

@@ -54,15 +54,12 @@ XOptimizer * XWorkerUpdate::GetOptimizer()
 
 /* 
 update a parameter of a model 
->> jobQueue - the queue that place the jobs called here
 >> model - the model that we want to update (on the server side)
->> members - models that would share the updated parameters
 >> pid - the parameter index
 >> optimizer - the optimizer
->> broadcaster - the worker that would broadcast the new parameter to members
 */
-void XWorkerUpdate::UpdateParameter(XQueue * jobQueue, XModel * server, XList * members, int pid,
-                                    XOptimizer * optimizer, XWorkerBroadcast * broadcaster)
+void XWorkerUpdate::UpdateParameter(XModel * server, int pid,
+                                    XOptimizer * optimizer)
 {
 
     CheckNTErrors(server->params[pid].flag == PARAM_STATE_COLLECTED, "The state of the parameter is wrong!");
@@ -77,10 +74,6 @@ void XWorkerUpdate::UpdateParameter(XQueue * jobQueue, XModel * server, XList *
 
     /* set the flag */
     server->params[pid].flag = PARAM_STATE_UPDATED;
-
-    /* broadcast the new parameter to other models (in anotehr worker/thread) */
-    broadcaster->AddJobBroadcastSingle(jobQueue, server, members, pid);
-    broadcaster->AddJobEnqueueFinished();
 }
 
 /* 
@@ -91,60 +84,45 @@ void XWorkerUpdate::Update(XList * args)
 {
     int paramCount = 0;
 
-    CheckNTErrors(args != NULL && args->count >= 6, "Illegal argument list!");
+    CheckNTErrors(args != NULL && args->count >= 4, "Illegal argument list!");
 
     XWorkerUpdate * updater = (XWorkerUpdate*)args->GetItem(paramCount++);
-    XQueue * jobQueue = (XQueue*)args->GetItem(paramCount++);
     XModel * server = (XModel*)args->GetItem(paramCount++);
-    int memNum = args->GetInt(paramCount++);
-
-    XList members;
-    for (int i = 0; i < memNum; i++) {
-        XModel * member = (XModel*)args->GetItem(paramCount++);
-        members.Add(member);
-    }
 
     int pid = args->GetInt(paramCount++);
     XOptimizer * optimizer = (XOptimizer*)args->GetItem(paramCount++);
-    XWorkerBroadcast * broadcaster = (XWorkerBroadcast*)args->GetItem(paramCount++);
 
     if(updater != NULL)
-        updater->UpdateParameter(jobQueue, server, &members, pid, optimizer, broadcaster);
+        updater->UpdateParameter(server, pid, optimizer);
 }
 
 /* 
 add a new job of model update (for a parameter) 
 >> jobQueue - the queue for sub-jobs executed in the job
 >> model - the model that we want to update (on the server side)
->> members - models that would share the updated parameters
 >> pid - the parameter index
 >> optimizer - the optimizer
->> broadcaster - the worker that would broadcast the new parameter to members
 */
 bool XWorkerUpdate::AddJobUpdate(XQueue * jobQueue,
-                                 XModel * model, XList * members, int pid,
-                                 XOptimizer * optimizer, XWorkerBroadcast * broadcaster)
+                                 XModel * model, int pid,
+                                 XOptimizer * optimizer)
 {
     CheckNTErrors(model != NULL, "No input model!");
-    CheckNTErrors(members != NULL, "No member model list!");
     CheckNTErrors(optimizer != NULL, "No optimizer!");
-    CheckNTErrors(broadcaster != NULL, "No broadcaster!");
     CheckNTErrors(pid >= 0 && pid < model->paramNum, "Illegal parameter index!");
 
     XList args;
     args.Add(this);
-    args.Add(jobQueue);
     args.Add(model);
-    args.AddInt(members->count);
-    args.AddList(members);
     args.AddInt(pid);
     args.Add(optimizer);
-    args.Add(broadcaster);
+    
+    XQueue& queueRun = jobQueue != NULL ? *jobQueue : queue;
 
     if (isInstantRun)
         XWorkerUpdate::Update(&args);
     else
-        queue.EnqueueJob((void*)(char*)XWorkerUpdate::Update, &args);
+        queueRun.EnqueueJob((void*)(char*)XWorkerUpdate::Update, &args);
 
     return true;
 }

source/train/XWorkerUpdate.h

@@ -57,8 +57,7 @@ public:
     XOptimizer * GetOptimizer();
 
     /* update the parameter */
-    void UpdateParameter(XQueue * jobQueue, XModel * server, XList * members, int pid,
-                         XOptimizer * optimizer, XWorkerBroadcast * broadcaster);
+    void UpdateParameter(XModel * server, int pid, XOptimizer * optimizer);
 
 
     /* wrapper of UpdateParameter */
@@ -67,8 +66,7 @@ public:
 
 
     /* add a new job of model update (for a parameter) */
-    bool AddJobUpdate(XQueue * jobQueue, XModel * model, XList * members, int pid,
-                      XOptimizer * optimizer, XWorkerBroadcast * broadcaster);
+    bool AddJobUpdate(XQueue * jobQueue, XModel * model, int pid, XOptimizer * optimizer);
 };
 
 }