Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
305e53ad
Unverified
Commit
305e53ad
authored
Jan 23, 2026
by
Matt
Committed by
GitHub
Jan 23, 2026
Browse files
[Hardware][AMD][CI][Bugfix] Fix Kernels Attention Cache test (#32904)
Signed-off-by:
Matthew Wong
<
Matthew.Wong2@amd.com
>
parent
1cb4341f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/kernels/attention/test_cache.py
tests/kernels/attention/test_cache.py
+1
-1
No files found.
tests/kernels/attention/test_cache.py
View file @
305e53ad
...
...
@@ -242,7 +242,7 @@ def test_reshape_and_cache_flash(
value_cache_compact
=
permute_and_compact
(
value_cache
)
def
convert_fp8_local
(
output
,
input
,
scale
,
kv_dtype
):
fp8_input
=
input
.
view
(
torch
.
float8_e4m3fn
)
fp8_input
=
input
.
view
(
current_platform
.
fp8_dtype
()
)
if
scale
.
numel
()
==
1
:
# per-tensor
result
=
scaled_dequantize
(
fp8_input
.
flatten
(
0
,
2
),
scale
,
group_shape
=
None
,
out_dtype
=
output
.
dtype
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment