This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from c976a07 [RUNTIME] Add libbacktrace for backtraces with line numbers
(#7153)
add 45442ed [Relay][Training][Pass] Factor out first-order AD to a module
pass (#7677)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/transform/transform.py | 26 +-
src/relay/transforms/first_order_gradient.cc | 309 +++++++++++++++++++++
.../relay/op.h => src/relay/transforms/gradient.h | 33 ++-
.../{gradient.cc => higher_order_gradient.cc} | 274 +-----------------
4 files changed, 360 insertions(+), 282 deletions(-)
create mode 100644 src/relay/transforms/first_order_gradient.cc
copy include/tvm/relay/op.h => src/relay/transforms/gradient.h (57%)
rename src/relay/transforms/{gradient.cc => higher_order_gradient.cc} (64%)
