This is an automated email from the ASF dual-hosted git repository.
comaniac pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from cb395ff Disable pip cache when creating Docker images (#8575)
add 850abb0 [TOPI] Add transpose_a/b & dynamic shape support for batch
matmul (#8527)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 14 +-
python/tvm/relay/frontend/tensorflow.py | 23 +++-
python/tvm/relay/frontend/tensorflow_ops.py | 15 ++-
python/tvm/relay/op/_tensor_grad.py | 56 +++++++-
python/tvm/relay/op/nn/_nn.py | 24 ++--
python/tvm/relay/op/nn/nn.py | 26 ++--
python/tvm/relay/op/op_attrs.py | 5 +
python/tvm/relay/op/strategy/cuda.py | 15 ++-
python/tvm/relay/op/strategy/generic.py | 5 +-
python/tvm/topi/cuda/batch_matmul.py | 115 ++++++++++++++--
python/tvm/topi/cuda/tensorcore_alter_op.py | 11 +-
python/tvm/topi/nn/batch_matmul.py | 134 ++++++++++++------
python/tvm/topi/testing/batch_matmul.py | 17 ++-
python/tvm/topi/x86/batch_matmul.py | 149 ++++++++++++---------
src/relay/op/make_op.h | 2 +-
src/relay/op/nn/nn.cc | 26 ++--
src/relay/op/nn/nn.h | 47 +++----
src/relay/qnn/op/batch_matmul.cc | 10 +-
.../transforms/combine_parallel_batch_matmul.cc | 13 +-
src/relay/transforms/combine_parallel_dense.cc | 3 +-
tests/python/frontend/tensorflow/test_forward.py | 47 +++++--
tests/python/relay/test_any.py | 92 +++++++++++++
tests/python/relay/test_op_grad_level10.py | 20 ++-
tests/python/relay/test_op_level10.py | 81 ++---------
24 files changed, 673 insertions(+), 277 deletions(-)
