Skip to content

Commit 2e56421

Browse files
author
Siyuan Feng
authored
[DLight] Update Adreno GEMV Rules (#17016)
When reduction axis is small, it's not necessary to use rfactor. This PR updates the gemv rule to use rfactor only when the reduction axis is large enough.
1 parent 3b97658 commit 2e56421

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/dlight/gpu/gemv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -711,7 +711,7 @@ def apply(
711711
if LOAD_V_SHARED is False:
712712
LOAD_V_TILE = 1
713713

714-
if not isinstance(len_r, int):
714+
if not isinstance(len_r, int) or len_r < LOAD_V_TILE * TR * SCALE_PACK * DEC_PACK:
715715
return None
716716

717717
if not isinstance(len_s, int):

0 commit comments

Comments
 (0)