"vscode:/vscode.git/clone" did not exist on "9831aec49fd541a8fe6cb65f0c3a65b03eccffe0"
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(): ...@@ -228,7 +228,6 @@ if current_platform.is_rocm():
num_kv_heads, num_kv_heads,
BLOCK_SIZE: tl.constexpr, BLOCK_SIZE: tl.constexpr,
QUANT: tl.constexpr, QUANT: tl.constexpr,
IS_FNUZ: tl.constexpr,
): ):
tid = tl.program_id(0) tid = tl.program_id(0)
head_id = tl.program_id(1) head_id = tl.program_id(1)
...@@ -314,7 +313,6 @@ if current_platform.is_rocm(): ...@@ -314,7 +313,6 @@ if current_platform.is_rocm():
num_kv_heads, num_kv_heads,
BLOCK_SIZE=head_size, BLOCK_SIZE=head_size,
QUANT=QUANT, 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