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 2022/07/15 00:26:03 UTC
[tvm] branch last-successful updated (e084791852 -> 7661ba8910)

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

github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git


    from e084791852 [MetaSchedule] Add MultiLevelTilingTensorCore rule for auto-tensorization on CUDA (#12059)
     add 7661ba8910 [Collage] CollagePartition pass (#12086)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/__init__.py                       |   1 +
 .../{_ffi/_ctypes => relay/collage}/__init__.py    |   9 +-
 python/tvm/{arith => relay/collage}/_ffi_api.py    |   4 +-
 python/tvm/relay/collage/collage.py                | 146 +++++
 python/tvm/relay/transform/transform.py            |  23 +
 src/relay/collage/candidate_partition.cc           |   3 +-
 src/relay/collage/candidate_partition_index.cc     | 150 +++++
 src/relay/collage/candidate_partition_index.h      | 102 ++++
 src/relay/collage/collage_partitioner.cc           | 352 ++++++++++++
 .../collage/{cost.cc => collage_partitioner.h}     |  33 +-
 src/relay/collage/cost_estimator.cc                |   8 +-
 src/relay/collage/cost_estimator.h                 |   4 +-
 src/relay/collage/gather_partition_specs.cc        | 214 +++++++
 src/relay/collage/gather_partition_specs.h         |  71 +++
 src/relay/collage/priority_queue.h                 |  72 +++
 src/relay/collage/utils.cc                         |   2 +-
 src/runtime/vm/vm.cc                               |   8 +-
 tests/python/relay/test_pass_collage_partition.py  | 617 +++++++++++++++++++++
 18 files changed, 1788 insertions(+), 31 deletions(-)
 copy python/tvm/{_ffi/_ctypes => relay/collage}/__init__.py (83%)
 copy python/tvm/{arith => relay/collage}/_ffi_api.py (89%)
 create mode 100644 python/tvm/relay/collage/collage.py
 create mode 100644 src/relay/collage/candidate_partition_index.cc
 create mode 100644 src/relay/collage/candidate_partition_index.h
 create mode 100644 src/relay/collage/collage_partitioner.cc
 copy src/relay/collage/{cost.cc => collage_partitioner.h} (54%)
 create mode 100644 src/relay/collage/gather_partition_specs.cc
 create mode 100644 src/relay/collage/gather_partition_specs.h
 create mode 100644 src/relay/collage/priority_queue.h
 create mode 100644 tests/python/relay/test_pass_collage_partition.py