NVIDIA / NVIDIA/TensorRT-LLM

Force KV Cache Offload

Open
#3,130 8 comments 0 reactions 1 assignee View on GitHub

@Kefeng-Duan is already working on this.

Since Mar 27, 2025.

question triaged
Dominant language
Python
Stars
14.7k
Forks
2.8k
Avg merge
2d 23h
Merged PRs (30d)
489

Description

Hello, I am trying to run kv_cache_offload benchmark . My question is there a way to force trt to only use CPU for KV Cache? This example shows how to offload KV Cache to CPU when its overflows GPU memory.
I am trying to run an experiment to check how much throughput/ latency get effected if I keep the KV Cache on GPU vs offload it to CPU

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.