---
id: pr-tensorrt-llm-14000
repo: NVIDIA/TensorRT-LLM
pr: 14000
title: '[TRTLLM-12589][fix] Reset MoE A2A dispatch state on warmup OOM'
author: Barry-Delaney
date: '2026-05-15'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/14000
source_category: upstream-code
architectures:
- sm100
tags:
- moe
techniques: []
hardware_features: []
kernel_types:
- moe
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: 12b810542cbc15e35ddf99efed34f718fecc8e70
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- tensorrt_llm/_torch/distributed/moe_alltoall.py
- tensorrt_llm/_torch/modules/fused_moe/communication/nvlink_one_sided.py
- tensorrt_llm/_torch/pyexecutor/model_engine.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-14000
head_ref_oid: 08828dbba17171fef26ee7d291e7d937e4074365
---
# [TRTLLM-12589][fix] Reset MoE A2A dispatch state on warmup OOM

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [14000](https://github.com/NVIDIA/TensorRT-LLM/pull/14000)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `tensorrt_llm/_torch/distributed/moe_alltoall.py`
- `tensorrt_llm/_torch/modules/fused_moe/communication/nvlink_one_sided.py`
- `tensorrt_llm/_torch/pyexecutor/model_engine.py`

## Evidence Bundle

`evidence/pull-bundles/tensorrt-llm/gh-14000`
