如何知道在MPI上運行哪個核心進程？

Question

我目前正在開發一個項目，我需要知道當前在MPI上運行該進程的處理器的coreid？ MPI中有一個名爲的函數MPI_Get_processor_name（char * name，int * resultlen）。這隻給出了進程正在運行的節點的名稱。我想知道它正在運行的核心的ID？可能嗎？如果是這樣，任何人都可以給我這個代碼片段嗎？

謝謝

Answer 1

這裏給出了它們綁定的每個進程的coreids的代碼。這需要Hristo Iliev在前面的回答評論中提出的hlloc庫。

#include <stdio.h> 
    #include "mpi.h" 
    #include <hwloc.h> 

    int main(int argc, char* argv[]) 
    { 
     int rank, size; 
     cpu_set_t mask; 
     long num; 
     int proc_num(long num); 

     hwloc_topology_t topology; 
     hwloc_cpuset_t cpuset; 
     hwloc_obj_t obj; 


     MPI_Init(&argc, &argv); 
     MPI_Comm_rank(MPI_COMM_WORLD, &rank); 
     MPI_Comm_size(MPI_COMM_WORLD, &size); 

     hwloc_topology_init (&topology); 
     hwloc_topology_load (topology); 

     hwloc_bitmap_t set = hwloc_bitmap_alloc(); 
     hwloc_obj_t pu; 
     int err; 

     err = hwloc_get_proc_cpubind(topology, getpid(), set, HWLOC_CPUBIND_PROCESS); 
     if (err) { 
     printf ("Error Cannot find\n"), exit(1); 
     } 

     pu = hwloc_get_pu_obj_by_os_index(topology, hwloc_bitmap_first(set)); 
     printf ("Hello World, I am %d and pid: %d coreid:%d\n",rank,getpid(),hwloc_bitmap_first(set)); 

     int my_coreid = hwloc_bitmap_first(set); 
     int all_coreid[size]; 
     hwloc_bitmap_free(set); 
     hwloc_topology_destroy(topology); 
     MPI_Finalize(); 
     return 0; 

} 

Answer 2

你的問題假定每個MPI進程運行綁定到單個CPU核心。這不是許多集羣MPI實施的默認行爲。例如，Open MPI具有必要的綁定機制，但必須使用--bind-to-core或--bind-to-socket選項明確啓用它。另一方面，現代英特爾MPI版本由於性能原因默認啓用綁定。由於這種差異，大多數集羣MPI實現MPI_GET_PROCESSOR_NAME只是簡單地返回執行節點的主機名，因爲在一般情況下沒有可識別的特定處理器。

當每個進程運行綁定到一個核心時，綁定通常可以通過讀取進程的親和性掩碼來獲得。這取決於操作系統，但有些庫可以將其抽象出來，例如hwloc library（Open MPI的一部分，但作爲完全獨立的項目開發，因此可以自行使用）。在一般情況下讀取親和性掩碼也是可能的 - 當一個進程未被綁定時，親和性掩碼將簡單地匹配系統親緣性掩碼（即在所有處理器上允許執行）。

有些平臺的綁定是系統硬件工作的一部分，例如， IBM Blue Gene。在每個調用過程中，每個MPI過程都在唯一可識別的處理器上執行，並且MPI_Get_processor_name返回唯一的字符串值。