mm_mpe_prof.c

刚才是说明 现在是安装程序在 LINUX环境下进行编程的MPICH安装文件

    MM_PROF_FUNC(MM_OPEN_PORT);
    MM_PROF_FUNC(MM_CLOSE_PORT);
    MM_PROF_FUNC(MM_ACCEPT);
    MM_PROF_FUNC(MM_CONNECT);
    MM_PROF_FUNC(MM_SEND);
    MM_PROF_FUNC(MM_RECV);
    MM_PROF_FUNC(MM_CLOSE);
    MM_PROF_FUNC(MM_REQUEST_ALLOC);
    MM_PROF_FUNC(MM_REQUEST_FREE);
    MM_PROF_FUNC(MM_CAR_INIT);
    MM_PROF_FUNC(MM_CAR_FINALIZE);
    MM_PROF_FUNC(MM_CAR_ALLOC);
    MM_PROF_FUNC(MM_CAR_FREE);
    MM_PROF_FUNC(MM_VC_INIT);
    MM_PROF_FUNC(MM_VC_FINALIZE);
    MM_PROF_FUNC(MM_VC_FROM_COMMUNICATOR);
    MM_PROF_FUNC(MM_VC_FROM_CONTEXT);
    MM_PROF_FUNC(MM_VC_ALLOC);
    MM_PROF_FUNC(MM_VC_CONNECT_ALLOC);
    MM_PROF_FUNC(MM_VC_FREE);
    MM_PROF_FUNC(MM_CHOOSE_BUFFER);
    MM_PROF_FUNC(MM_RESET_CARS);
    MM_PROF_FUNC(MM_GET_BUFFERS_TMP);
    MM_PROF_FUNC(MM_RELEASE_BUFFERS_TMP);
    MM_PROF_FUNC(MM_GET_BUFFERS_VEC);
    MM_PROF_FUNC(VEC_BUFFER_INIT);
    MM_PROF_FUNC(TMP_BUFFER_INIT);
    MM_PROF_FUNC(SIMPLE_BUFFER_INIT);
    MM_PROF_FUNC(MM_POST_RECV);
    MM_PROF_FUNC(MM_POST_SEND);
    MM_PROF_FUNC(MM_POST_RNDV_DATA_SEND);
    MM_PROF_FUNC(MM_POST_RNDV_CLEAR_TO_SEND);
    MM_PROF_FUNC(MM_CQ_TEST);
    MM_PROF_FUNC(MM_CQ_WAIT);
    MM_PROF_FUNC(MM_CQ_ENQUEUE);
    MM_PROF_FUNC(MM_CREATE_POST_UNEX);
    MM_PROF_FUNC(MM_ENQUEUE_REQUEST_TO_SEND);
    MM_PROF_FUNC(CQ_HANDLE_READ_HEAD_CAR);
    MM_PROF_FUNC(CQ_HANDLE_READ_DATA_CAR);
    MM_PROF_FUNC(CQ_HANDLE_READ_CAR);
    MM_PROF_FUNC(CQ_HANDLE_WRITE_HEAD_CAR);
    MM_PROF_FUNC(CQ_HANDLE_WRITE_DATA_CAR);
    MM_PROF_FUNC(CQ_HANDLE_WRITE_CAR);

    /* xfer functions */
    MM_PROF_FUNC(XFER_INIT);
    MM_PROF_FUNC(XFER_RECV_OP);
    MM_PROF_FUNC(XFER_RECV_MOP_OP);
    MM_PROF_FUNC(XFER_RECV_FORWARD_OP);
    MM_PROF_FUNC(XFER_RECV_MOP_FORWARD_OP);
    MM_PROF_FUNC(XFER_FORWARD_OP);
    MM_PROF_FUNC(XFER_SEND_OP);
    MM_PROF_FUNC(XFER_REPLICATE_OP);
    MM_PROF_FUNC(XFER_START);

    /* method functions */
    MM_PROF_FUNC(TCP_INIT);
    MM_PROF_FUNC(TCP_FINALIZE);
    MM_PROF_FUNC(TCP_ACCEPT_CONNECTION);
    MM_PROF_FUNC(TCP_GET_BUSINESS_CARD);
    MM_PROF_FUNC(TCP_CAN_CONNECT);
    MM_PROF_FUNC(TCP_POST_CONNECT);
    MM_PROF_FUNC(TCP_POST_READ);
    MM_PROF_FUNC(TCP_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(TCP_POST_WRITE);
    MM_PROF_FUNC(TCP_MAKE_PROGRESS);
    MM_PROF_FUNC(TCP_CAR_ENQUEUE);
    MM_PROF_FUNC(TCP_CAR_DEQUEUE);
    MM_PROF_FUNC(TCP_CAR_DEQUEUE_WRITE);
    MM_PROF_FUNC(TCP_RESET_CAR);
    MM_PROF_FUNC(TCP_POST_READ_PKT);
    MM_PROF_FUNC(TCP_READ);
    MM_PROF_FUNC(TCP_WRITE);
    MM_PROF_FUNC(TCP_READ_SHM);
    MM_PROF_FUNC(TCP_READ_VIA);
    MM_PROF_FUNC(TCP_READ_VIA_RDMA);
    MM_PROF_FUNC(TCP_READ_VEC);
    MM_PROF_FUNC(TCP_READ_TMP);
    MM_PROF_FUNC(TCP_READ_CONNECTING);
    MM_PROF_FUNC(TCP_WRITE_SHM);
    MM_PROF_FUNC(TCP_WRITE_VIA);
    MM_PROF_FUNC(TCP_WRITE_VIA_RDMA);
    MM_PROF_FUNC(TCP_WRITE_VEC);
    MM_PROF_FUNC(TCP_WRITE_TMP);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_SHM);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_VIA);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_VIA_RDMA);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_VEC);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_TMP);
    MM_PROF_FUNC(TCP_WRITE_AGGRESSIVE);
    MM_PROF_FUNC(TCP_CAR_HEAD_ENQUEUE);
    MM_PROF_FUNC(TCP_SETUP_PACKET_CAR);
    MM_PROF_FUNC(TCP_UPDATE_CAR_NUM_WRITTEN);
    MM_PROF_FUNC(TCP_MERGE_UNEXPECTED_DATA);
    MM_PROF_FUNC(TCP_MERGE_SHM);
    MM_PROF_FUNC(TCP_MERGE_VIA);
    MM_PROF_FUNC(TCP_MERGE_VIA_RDMA);
    MM_PROF_FUNC(TCP_MERGE_VEC);
    MM_PROF_FUNC(TCP_MERGE_TMP);
    MM_PROF_FUNC(TCP_MERGE_SIMPLE);
    MM_PROF_FUNC(TCP_MERGE_WITH_POSTED);
    MM_PROF_FUNC(TCP_READ_HEADER);
    MM_PROF_FUNC(TCP_READ_DATA);
    MM_PROF_FUNC(TCP_READ_SIMPLE);
    MM_PROF_FUNC(TCP_WRITE_SIMPLE);
    MM_PROF_FUNC(TCP_STUFF_VECTOR_SIMPLE);
    MM_PROF_FUNC(FIND_IN_QUEUE);

    MM_PROF_FUNC(SHM_CAN_CONNECT);
    MM_PROF_FUNC(SHM_GET_BUSINESS_CARD);
    MM_PROF_FUNC(SHM_INIT);
    MM_PROF_FUNC(SHM_FINALIZE);
    MM_PROF_FUNC(SHM_MAKE_PROGRESS);
    MM_PROF_FUNC(SHM_ALLOC);
    MM_PROF_FUNC(SHM_FREE);
    MM_PROF_FUNC(SHM_GET_MEM_SYNC);
    MM_PROF_FUNC(SHM_RELEASE_MEM);
    MM_PROF_FUNC(SHM_POST_CONNECT);
    MM_PROF_FUNC(SHM_POST_READ);
    MM_PROF_FUNC(SHM_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(SHM_POST_WRITE);

    MM_PROF_FUNC(PACKER_CAR_ENQUEUE);
    MM_PROF_FUNC(PACKER_CAR_DEQUEUE);
    MM_PROF_FUNC(PACKER_INIT);
    MM_PROF_FUNC(PACKER_FINALIZE);
    MM_PROF_FUNC(PACKER_MAKE_PROGRESS);
    MM_PROF_FUNC(PACKER_POST_READ);
    MM_PROF_FUNC(PACKER_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(PACKER_POST_WRITE);
    MM_PROF_FUNC(PACKER_RESET_CAR);

    MM_PROF_FUNC(UNPACKER_CAR_ENQUEUE);
    MM_PROF_FUNC(UNPACKER_CAR_DEQUEUE);
    MM_PROF_FUNC(UNPACKER_INIT);
    MM_PROF_FUNC(UNPACKER_FINALIZE);
    MM_PROF_FUNC(UNPACKER_MAKE_PROGRESS);
    MM_PROF_FUNC(UNPACKER_WRITE_SHM);
    MM_PROF_FUNC(UNPACKER_WRITE_VIA);
    MM_PROF_FUNC(UNPACKER_WRITE_VIA_RDMA);
    MM_PROF_FUNC(UNPACKER_WRITE_VEC);
    MM_PROF_FUNC(UNPACKER_WRITE_TMP);
    MM_PROF_FUNC(UNPACKER_POST_READ);
    MM_PROF_FUNC(UNPACKER_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(UNPACKER_POST_WRITE);
    MM_PROF_FUNC(UNPACKER_RESET_CAR);
    MM_PROF_FUNC(UNPACKER_WRITE_SIMPLE);

    MM_PROF_FUNC(VIA_CAN_CONNECT);
    MM_PROF_FUNC(VIA_GET_BUSINESS_CARD);
    MM_PROF_FUNC(VIA_INIT);
    MM_PROF_FUNC(VIA_FINALIZE);
    MM_PROF_FUNC(VIA_MAKE_PROGRESS);
    MM_PROF_FUNC(VIA_POST_CONNECT);
    MM_PROF_FUNC(VIA_POST_READ);
    MM_PROF_FUNC(VIA_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(VIA_POST_WRITE);

    MM_PROF_FUNC(VIA_RDMA_CAN_CONNECT);
    MM_PROF_FUNC(VIA_RDMA_GET_BUSINESS_CARD);
    MM_PROF_FUNC(VIA_RDMA_INIT);
    MM_PROF_FUNC(VIA_RDMA_FINALIZE);
    MM_PROF_FUNC(VIA_RDMA_MAKE_PROGRESS);
    MM_PROF_FUNC(VIA_RDMA_POST_CONNECT);
    MM_PROF_FUNC(VIA_RDMA_POST_READ);
    MM_PROF_FUNC(VIA_RDMA_MERGE_WITH_UNEXPECTED);
    MM_PROF_FUNC(VIA_RDMA_POST_WRITE);
}

int prof_init(int rank, int size)
{
    int i;
    unsigned char r,g,b;

    g_prof_rank = rank;

    g_pDLOG = DLOG_InitLog(rank, size);
    if (g_pDLOG == NULL)
	return -1;

    for (i=0; i<MM_NUM_TIMER_STATES; i++)
    {
	g_timer_state[i].num_calls = 0;
	g_timer_state[i].in_id = DLOG_GetNextEvent(g_pDLOG);
	g_timer_state[i].out_id = DLOG_GetNextEvent(g_pDLOG);
	g_timer_state[i].color = random_color(&r, &g, &b);
	/*sprintf(g_timer_state[i].color_str, "%d %d %d", (int)r, (int)g, (int)b);*/
	random_X_color_string(g_timer_state[i].color_str);
    }
    init_state_strings();

    strncpy(g_prof_filename, "mpid_prof", 256);
    
    DLOG_EnableLogging(g_pDLOG);
    DLOG_LogCommID(g_pDLOG, (int)MPI_COMM_WORLD);

    return MPI_SUCCESS;
}

int prof_finalize()
{
    int i;
    
    printf( "Writing logfile.\n");fflush(stdout);
    for (i=0; i<MM_NUM_TIMER_STATES; i++) 
    {
	DLOG_DescribeState(g_pDLOG,
	    g_timer_state[i].in_id, 
	    g_timer_state[i].out_id, 
	    g_timer_state[i].name,
	    g_timer_state[i].color_str );
    }

    DLOG_DisableLogging(g_pDLOG);

    DLOG_FinishLog(g_pDLOG, g_prof_filename);

    printf("finished.\n");fflush(stdout);

    return MPI_SUCCESS;
}

#endif