This is an automated email from the ASF dual-hosted git repository.
jroesch pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from 751982e [FIX][VM] Fix relay vm optimize (#6322)
add 1a26a2e [Relay] Make check stricter by using Feature. Fixed multiple
bugs. (#6326)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/feature.h | 34 +++++++
include/tvm/relay/transform.h | 9 ++
python/tvm/relay/prelude.py | 16 +--
python/tvm/relay/transform/transform.py | 17 +++-
src/relay/analysis/feature.cc | 51 +++++++++-
src/relay/transforms/gradient.cc | 55 ++++++++---
src/relay/transforms/lazy_gradient_init.cc | 107 ++++++++-------------
src/relay/transforms/partial_eval.cc | 5 +-
src/relay/transforms/pass_util.h | 3 +-
src/relay/transforms/to_a_normal_form.cc | 62 ++++++------
src/relay/transforms/to_cps.cc | 3 +
tests/python/relay/test_analysis_feature.py | 2 -
tests/python/relay/test_op_grad_level10.py | 14 ++-
tests/python/relay/test_pass_gradient.py | 24 ++++-
tests/python/relay/test_pass_lazy_gradient_init.py | 18 ++++
tests/python/relay/test_pass_merge_composite.py | 16 +--
tests/python/relay/test_pass_partial_eval.py | 25 +----
17 files changed, 291 insertions(+), 170 deletions(-)
