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 c208a1f support adb-shell style cpp_rpc (#8223)
add 8e63486 [Relay] [Pass] Add mixed precision (e.g. FP16) model
conversion pass (#8069)
No new revisions were added by this update.
Summary of changes:
include/tvm/runtime/data_type.h | 15 +
python/tvm/relay/op/__init__.py | 1 +
python/tvm/relay/op/op.py | 33 +-
python/tvm/relay/transform/mixed_precision.py | 195 +++++++++++
python/tvm/relay/transform/transform.py | 35 +-
python/tvm/topi/nn/conv2d.py | 10 +-
src/relay/transforms/to_mixed_precision.cc | 455 ++++++++++++++++++++++++++
tests/python/frontend/mxnet/test_forward.py | 20 +-
tests/python/relay/test_to_mixed_precision.py | 446 +++++++++++++++++++++++++
9 files changed, 1189 insertions(+), 21 deletions(-)
create mode 100644 python/tvm/relay/transform/mixed_precision.py
create mode 100644 src/relay/transforms/to_mixed_precision.cc
create mode 100644 tests/python/relay/test_to_mixed_precision.py
