MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / mpi_barrier

Function mpi_barrier

tensorrt_llm/_utils.py:591–593  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

589
590
591def mpi_barrier():
592 if ENABLE_MULTI_DEVICE:
593 mpi_comm().Barrier()
594
595
596def local_mpi_barrier():

Callers 7

build_from_hfFunction · 0.90
build_and_run_tp2Function · 0.90
load_tokenizerFunction · 0.90
barrierMethod · 0.90
parallel_buildFunction · 0.90
_load_model_from_hfMethod · 0.85
task_wrapperMethod · 0.85

Calls 1

mpi_commFunction · 0.70

Tested by 1

build_and_run_tp2Function · 0.72