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
788f284b
Unverified
Commit
788f284b
authored
Feb 27, 2025
by
Michael Goin
Committed by
GitHub
Feb 27, 2025
Browse files
Fix test_block_fp8.py test for MoE (#13915)
Signed-off-by:
mgoin
<
mgoin64@gmail.com
>
parent
4b1d141f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/kernels/test_block_fp8.py
tests/kernels/test_block_fp8.py
+2
-2
No files found.
tests/kernels/test_block_fp8.py
View file @
788f284b
...
...
@@ -30,8 +30,8 @@ M_moe = [1, 7, 83, 512, 2048]
N_moe
=
[
4608
]
# [128, 4608, 13824]
K_moe
=
[
7168
]
# [256, 7168, 13824]
BLOCK_SIZE
=
[[
128
,
128
]]
E
=
[
256
]
# [8, 24, 128, 256]
TOP_KS
=
[
1
]
# [1, 2, 6]
E
=
[
8
,
24
]
# [8, 24, 128, 256]
TOP_KS
=
[
2
]
# [1, 2, 6]
OUT_DTYPES
=
[
torch
.
bfloat16
]
# [torch.float32, torch.half, torch.bfloat16]
SEEDS
=
[
0
]
...
...
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