有没有办法使用MPI推断同一节点上有多少个工作进程？

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

int main(int argc, char** argv) {
    MPI_Init(NULL, NULL);

    int world_size;
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);

    int world_rank;
    MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);

    // 获取同一节点上的工作进程数量
    int node_size;
    MPI_Comm node_comm;
    MPI_Comm_split_type(MPI_COMM_WORLD, MPI_COMM_TYPE_SHARED, 0, MPI_INFO_NULL, &node_comm);
    MPI_Comm_size(node_comm, &node_size);

    printf("总进程数：%d, 当前进程：%d, 同一节点工作进程数量：%d\n", world_size, world_rank, node_size);

    MPI_Finalize();
}