You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by ma...@apache.org on 2022/05/07 06:39:05 UTC

[tvm] branch main updated (bc7f45e765 -> a3d75ae535)

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


    from bc7f45e765 [AOT] Enable A-Normal Form in the AOT executor (#11091)
     add a3d75ae535 [Frontend][PyTorch] Add: Relay stft operator (#11190)

No new revisions were added by this update.

Summary of changes:
 include/tvm/relay/attrs/transform.h           |  22 ++++
 python/tvm/relay/frontend/pytorch.py          |  16 ++-
 python/tvm/relay/op/_transform.py             |  44 +++++++
 python/tvm/relay/op/strategy/cuda.py          |  11 ++
 python/tvm/relay/op/strategy/generic.py       |  33 ++++++
 python/tvm/relay/op/transform.py              |  60 ++++++++++
 python/tvm/topi/__init__.py                   |   1 +
 python/tvm/topi/cuda/__init__.py              |   1 +
 python/tvm/topi/cuda/stft.py                  | 135 ++++++++++++++++++++++
 python/tvm/topi/stft.py                       | 125 ++++++++++++++++++++
 src/relay/op/tensor/transform.cc              |  57 ++++++++++
 tests/python/frontend/pytorch/test_forward.py |  45 +++++++-
 tests/python/relay/test_op_level3.py          | 158 +++++++++++++++++++++++++-
 13 files changed, 701 insertions(+), 7 deletions(-)
 create mode 100644 python/tvm/topi/cuda/stft.py
 create mode 100644 python/tvm/topi/stft.py