bodo.get_size

bodo.get_size() Get the total number of processes.

Example Usage

Save following code in get_rank.py file and run with multiple cores.

import bodo

@bodo.jit
def run_in_parallel():
    # some work only on rank 0
    if bodo.get_rank() == 0:
        print("rank 0 done")

    # some work on every process
    print("rank", bodo.get_rank(), "here")
    print("total ranks:", bodo.get_size())
run_in_parallel()

BODO_NUM_WORKERS=4 python get_rank.py

Output

rank 0 done
rank 0 here
total ranks: 4
rank 1 here
total ranks: 4
rank 2 here
total ranks: 4
rank 3 here
total ranks: 4