This is an automated email from the ASF dual-hosted git repository.
dickjc123 pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 16fed6e [FEATURE] add oneDNN support for numpy transpose (#20419)
add 36ed5e0 Port convolutions to cuDNN v8 API (#20635)
No new revisions were added by this update.
Summary of changes:
ci/docker/Dockerfile.build.centos7 | 2 +-
ci/docker/Dockerfile.build.ubuntu | 1 +
docs/static_site/src/pages/api/faq/env_var.md | 47 ++
include/mxnet/storage.h | 7 +-
src/common/cuda/cudnn_cxx.cc | 333 +++++++++
src/common/cuda/cudnn_cxx.h | 320 +++++++++
src/common/cuda/utils.h | 12 +-
src/operator/cudnn_ops.cc | 764 +++++++++++++++++++++
src/operator/cudnn_ops.h | 255 +++++++
src/operator/nn/convolution.cu | 175 +++--
src/operator/nn/cudnn/cudnn_batch_norm.cu | 1 -
src/operator/nn/cudnn/cudnn_batch_norm.h | 1 -
src/operator/nn/cudnn/cudnn_convolution-inl.h | 831 -----------------------
src/operator/nn/cudnn/cudnn_deconvolution-inl.h | 852 ------------------------
src/operator/nn/deconvolution.cu | 160 ++---
src/storage/cpu_device_storage.h | 5 +-
src/storage/cpu_shared_storage_manager.h | 4 +-
src/storage/gpu_device_storage.h | 16 +-
src/storage/naive_storage_manager.h | 6 +-
src/storage/pinned_memory_storage.h | 4 +-
src/storage/pooled_storage_manager.h | 20 +-
src/storage/storage.cc | 9 +-
src/storage/storage_manager.h | 3 +-
tests/python/gpu/test_gluon_model_zoo_gpu.py | 3 +-
tests/python/unittest/test_gluon.py | 3 +-
25 files changed, 1925 insertions(+), 1909 deletions(-)
create mode 100644 src/common/cuda/cudnn_cxx.cc
create mode 100644 src/common/cuda/cudnn_cxx.h
create mode 100644 src/operator/cudnn_ops.cc
create mode 100644 src/operator/cudnn_ops.h
delete mode 100644 src/operator/nn/cudnn/cudnn_convolution-inl.h
delete mode 100644 src/operator/nn/cudnn/cudnn_deconvolution-inl.h