Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "d886c26d4d4fef7d079696beb4ece1cfb4b008a8"
Unverified Commit 5e25b122 authored by Thomas Parnell's avatar Thomas Parnell Committed by GitHub
Browse files

[Kernel] [Mamba] Remove BLOCK_H=1 from list of tuneable configurations for... 


[Kernel] [Mamba] Remove BLOCK_H=1 from list of tuneable configurations for `_chunk_cumsum_fwd_kernel` (#25197)
Signed-off-by: default avatarThomas Parnell <tpa@zurich.ibm.com>
Co-authored-by: default avatarChih-Chieh-Yang <chih.chieh.yang@ibm.com>
parent c85d75cf
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
vllm/model_executor/layers/mamba/ops/ssd_chunk_state.py
View file @ 5e25b122
...@@ -17,7 +17,6 @@ from .mamba_ssm import softplus ...@@ -17,7 +17,6 @@ from .mamba_ssm import softplus
@triton.autotune( @triton.autotune(
configs=[ configs=[
triton.Config({'BLOCK_SIZE_H': 1}),
triton.Config({'BLOCK_SIZE_H': 2}), triton.Config({'BLOCK_SIZE_H': 2}),
triton.Config({'BLOCK_SIZE_H': 4}), triton.Config({'BLOCK_SIZE_H': 4}),
triton.Config({'BLOCK_SIZE_H': 8}), triton.Config({'BLOCK_SIZE_H': 8}),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment