MCPcopy Index your code
hub / github.com/NVIDIA/TensorRT-LLM / set_mpi_session_cpp

Function set_mpi_session_cpp

tensorrt_llm/llmapi/mpi_session.py:81–85  ·  view source on GitHub ↗
(comm)

Source from the content-addressed store, hash-verified

79
80
81def set_mpi_session_cpp(comm):
82 if ENABLE_MULTI_DEVICE:
83 comm_fortran = comm.py2f()
84 from tensorrt_llm.bindings import MpiComm
85 MpiComm.set_raw_mpi_session_by_fortran_handle(comm_fortran)
86
87
88class MpiSession(abc.ABC):

Callers 1

worker_mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected