You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2022/09/22 23:06:03 UTC
[tvm] branch last-successful updated (195ae72b5c -> 4e783a6087)
This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 195ae72b5c [TOPI] Fix dtype legalize logic for CPU dot product instruction (#12865)
add 86f9580498 [Relay] Fix handling of TransfromLayout in TE compiler cache (#12874)
add 4e783a6087 [TOPI] Add layer norm operator (#12864)
No new revisions were added by this update.
Summary of changes:
include/tvm/topi/nn/layer_norm.h | 117 ++++++++++++++++
include/tvm/topi/reduction.h | 23 ++++
python/tvm/topi/nn/__init__.py | 1 +
.../tvm/topi/{vision/reorg.py => nn/layer_norm.py} | 34 ++---
python/tvm/topi/testing/__init__.py | 1 +
.../{gather_python.py => layer_norm_python.py} | 40 +++---
src/relay/backend/te_compiler_cache.cc | 2 +-
src/tir/schedule/primitive/reduction.cc | 9 ++
src/topi/nn.cc | 6 +
tests/python/topi/python/test_topi_layer_norm.py | 62 +++++++++
...t_tir_transform_lower_cross_thread_reduction.py | 149 +++++++++++++++++++++
11 files changed, 411 insertions(+), 33 deletions(-)
create mode 100644 include/tvm/topi/nn/layer_norm.h
copy python/tvm/topi/{vision/reorg.py => nn/layer_norm.py} (54%)
copy python/tvm/topi/testing/{gather_python.py => layer_norm_python.py} (52%)
create mode 100644 tests/python/topi/python/test_topi_layer_norm.py