Home Research Papers Posters Manuals Talks Download People Help Internal

arch/util/machine-commthd-util.c

Go to the documentation of this file.

/* This file is considered be used inside the machine layer, not to be used separately */
#if CMK_SMP && CMK_LEVERAGE_COMMTHREAD

/* This msg buffer pool is only created on comm thread, and it's multi-producer-single-consumer */
CsvDeclare(PCQueue, notifyCommThdMsgBuffer);
CpvDeclare(int, notifyCommThdHdlr);

static void commThdHandleNotification(CmiNotifyCommThdMsg *msg){
    msg->fn(msg->numParams, msg->params);
    if(!msg->toKeep) CmiFreeNotifyCommThdMsg(msg);
}

/* Should be called in ConverseRunPE after ConverseCommonInit */
void CmiInitNotifyCommThdScheme(){
    CpvInitialize(int, notifyCommThdHdlr);
    CpvAccess(notifyCommThdHdlr) = CmiRegisterHandler((CmiHandler)commThdHandleNotification);;
    /* init the msg buffer */
    if(CmiMyRank() == CmiMyNodeSize()){
        int i;
        CsvAccess(notifyCommThdMsgBuffer) = PCQueueCreate();
        PCQueue q = CsvAccess(notifyCommThdMsgBuffer);
        /* create init buffer of 16 msgs */
        for(i=0; i<16; i++) {
            CmiNotifyCommThdMsg *one = (CmiNotifyCommThdMsg *)malloc(sizeof(CmiNotifyCommThdMsg));
            CmiSetHandler(one, CpvAccess(notifyCommThdHdlr));
            CmiBecomeImmediate(one);
            PCQueuePush(q, (char *)one);
        }
    }
    CmiNodeAllBarrier();
}

/* ============ Beginning of implementation for user APIs ============= */
CmiNotifyCommThdMsg *CmiCreateNotifyCommThdMsg(CmiCommThdFnPtr fn, int numParams, void *params, int toKeep){
    CmiNotifyCommThdMsg *one = (CmiNotifyCommThdMsg *)PCQueuePop(CsvAccess(notifyCommThdMsgBuffer));
    if(one == NULL) {
        one = (CmiNotifyCommThdMsg *)malloc(sizeof(CmiNotifyCommThdMsg));
        CmiSetHandler(one, CpvAccess(notifyCommThdHdlr));
        CmiBecomeImmediate(one);
    }
    one->fn = fn;
    one->numParams = numParams;
    one->params = params;
    one->toKeep = toKeep;
    return one;
}

void CmiFreeNotifyCommThdMsg(CmiNotifyCommThdMsg *msg){
    free(msg->params);
    /* Recycle the msg */
    PCQueuePush(CsvAccess(notifyCommThdMsgBuffer), (char *)msg);
}

/* Note: the "msg" has to be created from function call "CmiCreateNotifyCommThdMsg" */
void CmiResetNotifyCommThdMsg(CmiNotifyCommThdMsg *msg, CmiCommThdFnPtr fn, int numParams, void *params, int toKeep){
    msg->fn = fn;
    msg->numParams = numParams;
    msg->params = params;
    msg->toKeep = toKeep;
}

void CmiNotifyCommThd(CmiNotifyCommThdMsg *msg){
   CmiPushImmediateMsg(msg);
}
/* ============ End of implementation for user APIs ============= */
#endif

---