apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.46k stars 3.41k forks source link

[DLight] Update Adreno GEMV Rules #17016

Closed Hzfengsy closed 2 months ago

Hzfengsy commented 2 months ago

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.