Skip to content

Enhance GPU support in lm_eval script for FP8 attention and KV cache handling - #2619

Open
chensuyue wants to merge 1 commit into
mainfrom
suyue/fix-bench
Open

chensuyue wants to merge 1 commit into
mainfrom
suyue/fix-bench

Conversation

@chensuyue

@chensuyue chensuyue commented Oct 20, 2026 •

Copy link
Copy Markdown
Contributor

This pull request updates the logic in the benchmark/lm_eval/run_lm_eval.sh script to improve how attention backend and quantization settings are configured for different GPU architectures and evaluation formats. The changes make the script more robust and flexible when running benchmarks on various GPUs, especially in handling FP8 and QDQ plugin scenarios.

Improvements to attention backend and quantization logic:

  • The script now detects the GPU compute capability and applies the QDQ plugin and Marlin optimizations only on GPUs other than SM 10.x, moving this logic earlier and making it more robust. [1] [2]
  • The attention backend (ATTN_BACKEND) is now resolved based on whether FP8 is used for attention or KV cache, with clearer precedence and configuration.
  • The script ensures that the correct environment variables and CLI arguments are set for the chosen backend, handling both TRITON_ATTN and FLASHINFER cases, and only quantizes Q when attention runs in FP8.
  • Redundant and misplaced code for QDQ plugin activation was removed from the end of the script, as this logic is now handled earlier.

@chensuyue
chensuyue requested a review from xin3he October 20, 2026 14:42
…handling

Signed-off-by: chensuyue <suyue.chen@intel.com>

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant