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

Function current_all_reduce_helper

tensorrt_llm/plugin/plugin.py:771–774  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

769
770
771def current_all_reduce_helper():
772 global custom_all_reduce_helper
773 assert custom_all_reduce_helper is not None, "You must call `init_all_reduce_helper` first"
774 return custom_all_reduce_helper

Callers 11

test_allreduceMethod · 0.90
test_allreduceMethod · 0.90
test_pp_reduce_scatterMethod · 0.90
test_allreduceMethod · 0.90
prepare_inputsMethod · 0.90
prepare_inputsMethod · 0.90
allreduceFunction · 0.85
prepare_basic_inputsMethod · 0.85
prepare_inputsMethod · 0.85
prepare_inputsMethod · 0.85
prepare_inputsMethod · 0.85

Calls

no outgoing calls

Tested by 4

test_allreduceMethod · 0.72
test_allreduceMethod · 0.72
test_pp_reduce_scatterMethod · 0.72
test_allreduceMethod · 0.72