This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 432ccfa7a5 [Relax][PyTorch] Add support for and_, lshift, min, or_,
rshift, xor ops (#17668)
add aeadc31c97 [Docker] Use Torch GPU on gpu device (#17676)
add 1df8f9f4d7 [skip ci][CI] Update github tvmbot (#17675)
add 3c2a1abd65 [Dlight][CPU] Add CPU Backend Support for GEMV Optimization
(#17663)
No new revisions were added by this update.
Summary of changes:
ci/scripts/github/github_tvmbot.py | 4 -
docker/Dockerfile.ci_gpu | 2 +-
docker/install/ubuntu_install_onnx.sh | 45 +-
python/tvm/dlight/__init__.py | 9 +-
python/tvm/dlight/{base => analysis}/__init__.py | 10 +-
.../analysis.py => analysis/common_analysis.py} | 0
python/tvm/dlight/analysis/gemv.py | 162 ++++++
python/tvm/dlight/base/__init__.py | 16 +-
python/tvm/dlight/base/common_schedules.py | 2 +-
python/tvm/dlight/{gpu => base}/utils.py | 0
python/tvm/{_ffi/_cy2 => dlight/cpu}/__init__.py | 5 +-
python/tvm/dlight/{gpu => cpu}/base.py | 8 +-
python/tvm/dlight/cpu/gemv.py | 132 +++++
python/tvm/dlight/gpu/fallback.py | 7 +-
python/tvm/dlight/gpu/gemv.py | 132 +----
python/tvm/dlight/gpu/general_reduction.py | 3 +-
python/tvm/dlight/gpu/low_batch_gemv.py | 5 +-
python/tvm/dlight/gpu/matmul.py | 13 +-
python/tvm/dlight/gpu/reduction.py | 7 +-
python/tvm/dlight/gpu/rmsnorm.py | 4 +-
python/tvm/dlight/gpu/transpose.py | 7 +-
python/tvm/relax/frontend/nn/llm/tree_attn.py | 2 +
tests/python/dlight/test_cpu_gemv.py | 595 +++++++++++++++++++++
23 files changed, 979 insertions(+), 191 deletions(-)
copy python/tvm/dlight/{base => analysis}/__init__.py (84%)
rename python/tvm/dlight/{base/analysis.py => analysis/common_analysis.py}
(100%)
create mode 100644 python/tvm/dlight/analysis/gemv.py
rename python/tvm/dlight/{gpu => base}/utils.py (100%)
copy python/tvm/{_ffi/_cy2 => dlight/cpu}/__init__.py (93%)
copy python/tvm/dlight/{gpu => cpu}/base.py (81%)
create mode 100644 python/tvm/dlight/cpu/gemv.py
create mode 100644 tests/python/dlight/test_cpu_gemv.py
