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

Method has_moe_ep

tensorrt_llm/mapping.py:312–313  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

310 return self.moe_tp_size > 1
311
312 def has_moe_ep(self):
313 return self.moe_ep_size > 1
314
315 def pp_layers(self, num_layers: int) -> List[int]:
316 if self.pp_partition is not None:

Callers 8

__init__Method · 0.80
forward_expertsMethod · 0.80
convert_layerFunction · 0.80
convert_layerFunction · 0.80
load_weights_from_hf_safetensorsFunction · 0.80
convert_layerFunction · 0.80
load_weights_from_hf_modelFunction · 0.80
convert_hf_qwenFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected