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