PETSc和MPI COMM世界正確使用

Question

嗨我有一個用C++編寫的程序，其中一個或兩個函數使用PETSc。

當我調用這些功能，同時使用的PETSc

1) a)change all MPI_COMM_WORLD to PETSC_COMM_WORLD for my entire program 
    b)change MPI_Init to PetscInitialize 

這將是正確的方法..

2）讓PETSc創建一個單獨的通訊世界，傳遞給了函數

MPI_Init(&argc, &argv); 
    .. 
    //some many line of code 
    .. 

    MPI_Comm_split(MPI_COMM_WORLD, rank, 0, &PETSC_COMM_WORLD); 

    petsc_function(PETSC_COMM_WORLD,..... 

    //some how reverting the comm_split here to continue as normal so all the non petsc functions work 

..

3）或可我只是通過只使用MPI_得到具有Petsc功能的COMM_WORLD ..

4）還是有另一種方法我不知道？

Answer 1

您需要先撥打PetscInitialize。您可以用它替換電話MPI_Init。

沒有必要設置PETSC_COMM_WORLD，除非您只想在排名的子集上使用PETSc。如果您自己沒有設置PETSC_COMM_WORLD，它將自動由PetscInitialize設置爲MPI_COMM_WORLD的副本。

然後，當您打電話給PETSc功能時，您可以使用PETSC_COMM_WORLD。在程序結束時呼叫PetscFinalize，除非您單獨手動調用MPI_Init而不是單獨使用PetscInitialize，否則將爲您撥打MPI_Finalize。

在程序的其餘部分沒有必要將MPI_COMM_WORLD更改爲PETSC_COMM_WORLD。