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

Function mpi_disabled

tensorrt_llm/_utils.py:535–537  ·  view source on GitHub ↗

True if TLLM_DISABLE_MPI is set to "1", False otherwise.

()

Source from the content-addressed store, hash-verified

533
534
535def mpi_disabled() -> bool:
536 """True if TLLM_DISABLE_MPI is set to "1", False otherwise."""
537 return os.environ.get("TLLM_DISABLE_MPI") == "1"
538
539
540def mpi_rank():

Callers 15

run_disaggregated_testFunction · 0.90
conftest.pyFile · 0.90
util.pyFile · 0.90
__new__Method · 0.90
__init__Method · 0.90
__init__Method · 0.90
__init__Method · 0.90
_allgatherFunction · 0.90
allgatherFunction · 0.90
cp_allgatherFunction · 0.90
reducescatterFunction · 0.90
__init__Method · 0.90

Calls 1

getMethod · 0.45

Tested by 1

run_disaggregated_testFunction · 0.72