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 174d09e [TE][TIR] Implement layout transformations, non-flat memory
buffers (#9727)
add 12f213a [BYOC-DNNL] Support DNNL optimal layout (#10421)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/contrib/dnnl.py | 246 +++++++++++++---
src/relay/backend/contrib/dnnl/codegen.cc | 42 ++-
src/relay/backend/contrib/dnnl/query_layout.cc | 378 +++++++++++++++++++++++++
src/relay/op/nn/convolution.cc | 48 +++-
src/runtime/contrib/dnnl/dnnl_json_runtime.cc | 291 +++++++++++++------
src/tir/ir/data_layout.cc | 1 -
tests/python/contrib/test_dnnl.py | 158 +++++++++--
7 files changed, 990 insertions(+), 174 deletions(-)
create mode 100755 src/relay/backend/contrib/dnnl/query_layout.cc
