• Yuting Jiang's avatar
    Benchmarks: micro benchmarks - add --set_ib_devices option to auto-select IB... · c65ae567
    Yuting Jiang authored
    Benchmarks: micro benchmarks - add --set_ib_devices option to auto-select IB device by MPI local rank in ib validation (#733)
    
    **Description**
    add --set_ib_devices option to auto-select IB device by MPI local rank 
    
    
    **Major Revision**
    - Add a new CLI flag --set_ib_devices to automatically select irregular
    IB devices based on the MPI local rank.
    - When enabled, the benchmark queries available IB devices via
    network.get_ib_devices() and selects the device corresponding to
    OMPI_COMM_WORLD_LOCAL_RANK.
    - Fall back to existing --ib_dev behavior when the flag is not provided.
    
    **Minor Revision**
    - Add an env in network.get_ib_devices() to allow user to set the device
    name
    c65ae567
ib_validation_performance.py 18.2 KB