From 136872e8935a42a0a766879186609b64745180b4 Mon Sep 17 00:00:00 2001 From: Yanchao Lu Date: Mon, 24 Aug 2026 22:12:45 +0800 Subject: [PATCH] [None][fix] Restore CpType import for HELIX KV cache Signed-off-by: Yanchao Lu --- tensorrt_llm/_torch/pyexecutor/_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_llm/_torch/pyexecutor/_util.py b/tensorrt_llm/_torch/pyexecutor/_util.py index ca92ae05522e..95c92473efa5 100644 --- a/tensorrt_llm/_torch/pyexecutor/_util.py +++ b/tensorrt_llm/_torch/pyexecutor/_util.py @@ -40,7 +40,7 @@ from tensorrt_llm._torch.peft.lora.manager import (load_torch_lora, supports_native_fp8_lora) from tensorrt_llm.logger import logger -from tensorrt_llm.mapping import Mapping +from tensorrt_llm.mapping import CpType, Mapping from ..attention_backend import get_sparse_attn_kv_cache_manager from ..hostfunc import set_low_latency_dispatch