Skip to content

Commit e5adafe

Browse files
authored
[Fix] gather_gemv small fix (#452)
1 parent 30c6f6d commit e5adafe

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tritonbench/operators/gather_gemv/operator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(
3838
):
3939
super().__init__(tb_args, extra_args)
4040

41-
@register_benchmark(baseline=True)
41+
@register_benchmark()
4242
def test_0(self, p1, p2, p3) -> Callable:
4343
return lambda: triton_test_0(p1, p2, p3)
4444

tritonbench/operators/gather_gemv/triton_gather_gemv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,4 +126,4 @@ def triton_gemv_0(arg0_1, arg1_1, arg2_1):
126126

127127
grid = lambda META: (triton.cdiv(2 * S, META["XBLOCK"]),)
128128
triton_red_fused_mv_0[grid](arg1_1, arg0_1, arg2_1, buf1, xnumel, rnumel)
129-
return (reinterpret_tensor(buf1, (2, S), (S, 1), 0),)
129+
return reinterpret_tensor(buf1, (2, S), (S, 1), 0)

0 commit comments

Comments
 (0)