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

Method size

tensorrt_llm/_torch/device_mesh.py:40–41  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

38
39 @staticmethod
40 def size():
41 return 1
42
43
44class DeviceMeshTopologyImpl(_MappingBaseForTypeCheck):

Callers 15

pixel_shuffleMethod · 0.45
build_qwen2_audio_engineFunction · 0.45
attn_forwardFunction · 0.45
_maybe_add_infoMethod · 0.45
_get_input_sizesMethod · 0.45
_optimization_profilesMethod · 0.45
post_load_weightsMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
forward_context_with_chunked_prefillMethod · 0.45
forward_absorption_generationMethod · 0.45
_tensor_to_weightFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected