kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5474119][fix] Re-enable test (#8809)

Browse Source 
Signed-off-by: Dongfeng Yu <dongfengy@nvidia.com>
This commit is contained in:
dongfengy 2025-10-31 10:17:58 -07:00 committed by GitHub
parent 6424f7e55f
commit 0edba5a7e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unittest/_torch/thop/parallel/test_moe.py
View File

@ -1840,7 +1840,7 @@ def test_moe_fp8_per_tensor_scale(num_tokens, expert_info, hidden_size,
reason="The kernel only supports Blackwell. Current SM is %d." %
getSMVersion(),
)
@pytest.mark.parametrize("num_tokens", [1, 1024])
@pytest.mark.parametrize("num_tokens", [1, 256, 1024])
@pytest.mark.parametrize("hidden_size", [512])
@pytest.mark.parametrize("intermediate_size", [512])
@pytest.mark.parametrize(