[ROCm] Remove unused IS_FNUZ parameter from reshape_and_cache_shuffle_kernel (#39123)

Signed-off-by: Bortlesboat <bortstheboat@gmail.com>
This commit is contained in:
Andrew Barnes
2026-04-07 03:17:59 -04:00
committed by GitHub
parent a435e3108d
commit 5c35517a3e

View File

@@ -228,7 +228,6 @@ if current_platform.is_rocm():
num_kv_heads,
BLOCK_SIZE: tl.constexpr,
QUANT: tl.constexpr,
IS_FNUZ: tl.constexpr,
):
tid = tl.program_id(0)
head_id = tl.program_id(1)
@@ -314,7 +313,6 @@ if current_platform.is_rocm():
num_kv_heads,
BLOCK_SIZE=head_size,
QUANT=QUANT,
IS_FNUZ=current_platform.fp8_dtype() == torch.float8_e4m3fnuz,
)