This is an automated email from the ASF dual-hosted git repository.
sxjscience pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from a6fb7c2 Porting: Fixed setting attributes in reviewSubgraph (#19277)
add e67d53e Einsum cutensor GPU (#18921)
No new revisions were added by this update.
Summary of changes:
3rdparty/mshadow/CMakeLists.txt | 3 +
3rdparty/mshadow/mshadow/base.h | 11 +
3rdparty/mshadow/mshadow/stream_gpu-inl.h | 61 +-
CMakeLists.txt | 12 +
include/mxnet/libinfo.h | 4 +
src/common/cuda/utils.h | 6 +
src/operator/numpy/np_einsum_op-inl.h | 4 +
src/operator/numpy/np_einsum_op.cu | 892 ++++++++++++++++++++++++++++-
src/operator/numpy/np_einsum_path_op-inl.h | 91 ++-
tests/python/gpu/test_numpy_op.py | 159 +++++
10 files changed, 1239 insertions(+), 4 deletions(-)
create mode 100644 tests/python/gpu/test_numpy_op.py
