Unverified Commit 5c35517a authored by Andrew Barnes's avatar Andrew Barnes Committed by GitHub
Browse files

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


Signed-off-by: default avatarBortlesboat <bortstheboat@gmail.com>
parent a435e310
......@@ -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,
)
......
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