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
......@@ -17,7 +17,6 @@ from .mamba_ssm import softplus
@triton.autotune(
configs=[
triton.Config({'BLOCK_SIZE_H': 1}),
triton.Config({'BLOCK_SIZE_H': 2}),
triton.Config({'BLOCK_SIZE_H': 4}),
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