This is an automated email from the ASF dual-hosted git repository.
samskalicky pushed a change to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 5079c35 [v1.x] Backport Improve environment variable handling in
unittests (#18424) (#19173)
add 0fce381 [1.x][FEATURE] CUDA graphs support (#19142)
No new revisions were added by this update.
Summary of changes:
docs/static_site/src/pages/api/faq/env_var.md | 10 +
include/mxnet/op_attr_types.h | 13 +
src/executor/attach_op_execs_pass.cc | 64 +--
src/executor/cuda_graphs.h | 547 +++++++++++++++++++++++++
src/executor/exec_pass.h | 18 +
src/executor/graph_executor.cc | 40 +-
src/imperative/imperative_utils.h | 16 +-
src/operator/contrib/adamw.cu | 16 +
src/operator/nn/dropout.cu | 5 +
src/operator/numpy/linalg/np_eig.cu | 8 +
src/operator/numpy/linalg/np_eigvals.cu | 8 +
src/operator/numpy/linalg/np_norm_backward.cu | 5 +
src/operator/numpy/linalg/np_norm_forward.cu | 5 +
src/operator/numpy/np_boolean_mask_assign.cu | 8 +
src/operator/numpy/np_constraint_check.cu | 4 +
src/operator/numpy/np_percentile_op.cu | 4 +
src/operator/numpy/random/np_bernoulli_op.cu | 4 +
src/operator/numpy/random/np_exponential_op.cu | 4 +
src/operator/numpy/random/np_gamma_op.cu | 4 +
src/operator/numpy/random/np_multinomial_op.cu | 4 +
src/operator/numpy/random/np_normal_op.cu | 8 +
src/operator/numpy/random/np_pareto_op.cu | 4 +
src/operator/numpy/random/np_power_op.cu | 4 +
src/operator/numpy/random/np_rayleigh_op.cu | 4 +
src/operator/numpy/random/np_weibull_op.cu | 4 +
src/operator/tensor/indexing_op.cu | 4 +
tests/python/gpu/test_gluon_gpu.py | 112 ++++-
27 files changed, 884 insertions(+), 43 deletions(-)
create mode 100644 src/executor/cuda_graphs.h