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 2024/03/11 05:02:43 UTC

(tvm) branch nightly updated (40dd376375 -> 7ac03ca960)

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 40dd376375 [Unity][TIR] Clear struct info when specializing PrimFunc (#16584)
     add af82d97043 [Relax][Frontend][Onnx] support MaxPool1/2/3D and AveragePool1/2/3D (#16681)
     add 9b3621bf39 [CI] add merge_with_main in unity (#16661)
     add 7ab970d38e [Lint] Add check to prevent usage of #include <regex> (#16412)
     add c43fad1d60 [Relax] Implement StructInfoPattern for dataflow pattern matching (#16685)
     add 86c5df8923 [Runtime][RPC] Fix FreeObject in minrpc server (#16647)
     add d894334800 [COMMUNITY] Add new key for release signing (#16695)
     add f988cb48d1 [Bugfix][SLM] Produce well-formed Relax for nn.modules.KVCache (#16684)
     add dc7d6873ba [Runtime] PagedKVCache execute data copy on a separate stream (#16692)
     add 639a6e4f3c [Contrib] Support NDArray cache taking generator (#16693)
     add 7ac03ca960 [WEB] Initial support for asyncify (#16694)

No new revisions were added by this update.

Summary of changes:
 3rdparty/flashinfer                                |   2 +-
 KEYS                                               |  59 ++++
 ci/jenkins/unity_jenkinsfile.groovy                |  40 ++-
 include/tvm/relax/analysis.h                       |  22 ++
 include/tvm/relax/attrs/nn.h                       |  80 +++++
 include/tvm/relax/dataflow_pattern.h               |  27 ++
 include/tvm/relax/dataflow_pattern_functor.h       |   4 +
 python/tvm/contrib/emcc.py                         |   9 +-
 python/tvm/contrib/tvmjs.py                        |  43 ++-
 python/tvm/relax/dpl/pattern.py                    |  24 ++
 python/tvm/relax/frontend/nn/core.py               |  12 +-
 python/tvm/relax/frontend/nn/exporter.py           |   4 +-
 python/tvm/relax/frontend/nn/modules.py            |  27 +-
 python/tvm/relax/frontend/onnx/onnx_frontend.py    | 102 ++++--
 python/tvm/relax/op/_op_gradient.py                |   2 +
 python/tvm/relax/op/grad/grad.py                   |  24 +-
 python/tvm/relax/op/nn/__init__.py                 |   4 +
 python/tvm/relax/op/nn/nn.py                       | 354 ++++++++++++++++++++-
 python/tvm/relax/transform/legalize_ops/grad.py    |   2 +
 python/tvm/relax/transform/legalize_ops/nn.py      |  95 ++++++
 python/tvm/runtime/__init__.py                     |   2 +
 python/tvm/runtime/support.py                      |  69 ++++
 python/tvm/support.py                              |  44 ---
 python/tvm/topi/nn/pooling.py                      |   7 +-
 src/ir/transform.cc                                |   9 +-
 src/relax/analysis/struct_info_analysis.cc         | 200 ++++++++++++
 src/relax/ir/dataflow_matcher.cc                   |  55 ++++
 src/relax/ir/dataflow_matcher_impl.h               |  15 +
 src/relax/ir/dataflow_pattern.cc                   |  22 ++
 src/relax/ir/dataflow_pattern_functor.cc           |   4 +
 src/relax/op/nn/pooling.cc                         | 286 ++++++++++++++++-
 src/relax/op/nn/pooling.h                          |   6 +-
 src/relax/op/tensor/grad.cc                        |   8 +-
 src/relax/op/tensor/grad.h                         |   6 +-
 src/relax/transform/update_param_struct_info.cc    |   2 +-
 src/relay/backend/contrib/dnnl/codegen.cc          |   1 -
 src/relay/backend/contrib/dnnl/query_layout.cc     |  18 +-
 src/relay/backend/contrib/mrvl/codegen.cc          |   1 -
 src/runtime/c_runtime_api.cc                       |   1 -
 src/runtime/contrib/dnnl/dnnl_json_runtime.cc      |  63 ++--
 src/runtime/crt/common/crt_runtime_api.c           |   2 +
 src/runtime/minrpc/minrpc_server.h                 |   2 +
 .../curand/helper_cuda_kernels.h => regex.cc}      |  28 +-
 src/runtime/regex.h                                |  64 ++++
 src/runtime/relax_vm/paged_kv_cache.cc             | 161 ++++++----
 tests/lint/cpplint.sh                              |  10 +
 tests/python/relax/test_dataflow_pattern.py        | 169 ++++++++++
 tests/python/relax/test_frontend_nn_modules.py     |  10 +-
 tests/python/relax/test_frontend_onnx.py           | 251 +++++++++++----
 tests/python/relax/test_op_gradient_numeric.py     |  10 +-
 .../relax/test_transform_legalize_ops_grad.py      |   3 +-
 .../python/relax/test_transform_legalize_ops_nn.py |   7 +-
 tests/python/runtime/test_runtime_rpc.py           |   4 +
 web/Makefile                                       |   5 +-
 web/apps/node/example.js                           |   2 +-
 web/emcc/decorate_as_wasi.py                       |   1 +
 web/emcc/wasm_runtime.cc                           |   5 +
 web/emcc/webgpu_runtime.cc                         |   6 +-
 web/src/artifact_cache.ts                          |  46 ++-
 web/src/asyncify.ts                                | 227 +++++++++++++
 web/src/runtime.ts                                 |  77 ++++-
 web/src/support.ts                                 |  12 +
 web/tests/node/test_packed_func.js                 |  30 ++
 63 files changed, 2543 insertions(+), 344 deletions(-)
 create mode 100644 python/tvm/runtime/support.py
 copy src/runtime/{contrib/curand/helper_cuda_kernels.h => regex.cc} (61%)
 create mode 100644 src/runtime/regex.h
 create mode 100644 web/src/asyncify.ts