2013-11-27 64 views
3

我有以下MPI_DataType代碼返回MPI數據類型爲MPI收集

typedef struct resultset { 
    int rank, totalProcessed, bracket1, bracket2, bracket3, bracket4; 
    float bracket1percent, bracket2percent, bracket3percent, bracket4percent; 
} resultset; 

MPI_Datatype createResultType() 
{ 
    // Set-up the arguments for the type constructor 
    MPI_Datatype new_type; 

    int count = 2; 

    int blocklens[] = { 6, 4 }; 

    MPI_Aint indices[2]; 
    indices[0] = 0; 
    MPI_Type_extent(MPI_FLOAT, &indices[1]); 
    indices[1] *= 4; // There are 4 float 

    MPI_Datatype old_types[] = { MPI_INT, MPI_FLOAT }; 

    // Call the data type constructor 
    MPI_Type_struct(count, blocklens, indices, old_types, &new_type); 
    MPI_Type_commit(&new_type); 

    return new_type; 
} 

我嘗試使用下面的代碼從其他進程收到的數據。 rank是一個整數,它定義了進程的級別。

MPI_Datatype resType = createResultType(); 
if(rank != 0){ 
    MPI_Gather(&results, sizeof(resultset), resType, NULL, 1, resType, 0, MPI_COMM_WORLD); 
} 
else { 
    resultset* all_results = new resultset[numProcs]; 
    MPI_Gather(&results, sizeof(resultset), resType, all_results, sizeof(resultset), resType, 0, MPI_COMM_WORLD); 
} 

的問題是,我可以通過all_results不循環,使用這種方法:

for(int i = 0; i < numProcs; ++i){ 
    std::cout << all_results[i].rank << " processed " << all_results[i].totalProcessed <<std::endl; 
} 

我認爲問題在於我的數據類型,但我不知道。任何意見,將不勝感激。

編輯︰我得到了循環它的問題,現在我越來越不完整的數據。我在收集功能中將sizeof(resultset)更改爲1。現在我收到了正確的數據,除了最後兩個浮點數是-431602080而不是0.0022234327654之外。

MPI_Datatype resType = createResultType(); 
if(rank != 0){ 
    MPI_Gather(&results, 1, resType, NULL, 1, resType, 0, MPI_COMM_WORLD); 
} 
else { 
    resultset* all_results = new resultset[numProcs]; 
    MPI_Gather(&results, 1, resType, all_results, 1, resType, 0, MPI_COMM_WORLD); 
} 

回答

1

浮法需要是 「6」

int blocklens[] = { 6, 4 }; 

int blocklens[] = { 6, 6 };