Function mpi_broadcast

tensorrt_llm/_utils.py:601–602 · view source on GitHub ↗

(obj, root=0)

Source from the content-addressed store, hash-verified

599
600
601	def mpi_broadcast(obj, root=0):
602	return mpi_comm().bcast(obj, root) if global_mpi_size() > 1 else obj
603
604
605	def mpi_allgather(obj):

mainFunction · 0.90

_run_on_leaderMethod · 0.90

_download_hf_modelMethod · 0.85

from_dirMethod · 0.85

global_mpi_sizeFunction · 0.85

mpi_commFunction · 0.70

no test coverage detected