Address review comments (benchmark logs, ids, and design comments)

Signed-off-by: Yufeng Zhu <yzhu802@gatech.edu>
Signed-off-by: princepride <wangzhipeng628@gmail.com>
This commit is contained in:
Yufeng Zhu authored and princepride committed 2026-05-29 02:34:52 +00:00
1 parent e1834ab19b
commit 424fa3fcfc
6 files changed
+32 -7

No files matched your search

@@ -299,7 +299,7 @@ def _run_timing(
@pytest.mark.parametrize(
"image_num, K, N, dtype",
[(num, 3584, 4608, "w16a16_bf16") for num in _IMAGE_NUM],
ids=lambda val: "",
ids=[f"img{num}_K3584_N4608" for num in _IMAGE_NUM],
)
def test_mot_qkv_parallel(image_num: int, K: int, N: int, dtype: str, bias: bool):
dcfg = _parse_dtype(dtype)
@@ -384,7 +384,7 @@ def test_mot_qkv_parallel(image_num: int, K: int, N: int, dtype: str, bias: bool
@pytest.mark.parametrize(
"image_num, K, N, dtype",
[(num, 3584, 3584, "w16a16_bf16") for num in _IMAGE_NUM],
ids=lambda val: "",
ids=[f"img{num}_K3584_N3584" for num in _IMAGE_NUM],
)
def test_mot_o_proj(
image_num: int,