Skip to content

GitLab

Unverified Commit 6929f8b4 authored by Yang Chen's avatar Yang Chen Committed by GitHub
Browse files

[Misc] fixed nvfp4_moe test failures due to invalid kwargs (#21246) 


Signed-off-by: default avatarYang Chen <yangche@fb.com>
parent 32ec9e2f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/kernels/moe/test_nvfp4_moe.py
View file @ 6929f8b4
...@@ -93,11 +93,11 @@ def test_cutlass_fp4_moe_no_graph(m: int, n: int, k: int, e: int, topk: int, ...@@ -93,11 +93,11 @@ def test_cutlass_fp4_moe_no_graph(m: int, n: int, k: int, e: int, topk: int,
a1_gscale=a1_gs, a1_gscale=a1_gs,
w1_fp4=w1_q, w1_fp4=w1_q,
w1_blockscale=w1_blockscale, w1_blockscale=w1_blockscale,
w1_alphas=(1 / w1_gs), g1_alphas=(1 / w1_gs),
a2_gscale=a2_gs, a2_gscale=a2_gs,
w2_fp4=w2_q, w2_fp4=w2_q,
w2_blockscale=w2_blockscale, w2_blockscale=w2_blockscale,
w2_alphas=(1 / w2_gs), g2_alphas=(1 / w2_gs),
topk_weights=topk_weights, topk_weights=topk_weights,
topk_ids=topk_ids, topk_ids=topk_ids,
m=m, m=m,
......
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