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(-)