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

Function mpi_comm

tensorrt_llm/_utils.py:506–507  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

504
505
506def mpi_comm():
507 return comm
508
509
510local_comm = mpi_comm().Split_type(split_type=OMPI_COMM_TYPE_HOST)

Callers 15

create_executorMethod · 0.90
__init__Method · 0.90
__init__Method · 0.90
get_or_scale_allreduce_mnnvl_workspaceFunction · 0.90
broadcastMethod · 0.90
tp_commMethod · 0.90
pp_commMethod · 0.90
cp_commMethod · 0.90
allreduceMethod · 0.90
__init__Method · 0.90
parallel_buildFunction · 0.90
setup_engineMethod · 0.90

Calls

no outgoing calls

Tested by 1

create_executorMethod · 0.72