You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by pa...@apache.org on 2020/04/16 02:38:09 UTC
[incubator-mxnet] branch v1.x updated (2ccbcec -> 1afdfce)
This is an automated email from the ASF dual-hosted git repository.
patriczhao pushed a change to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 2ccbcec GPU gemms true fp16 (#17466) (#18023)
add 1afdfce [1.7] Backport MXNet Extension PRs (#17623, #17569, #17762) #18063 (#18069)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 15 +-
example/extensions/lib_custom_op/Makefile | 10 +-
example/extensions/lib_custom_op/relu_lib.cu | 90 ++++-
example/extensions/lib_custom_op/test_relu.py | 43 ++-
.../extensions/lib_custom_op/test_transposecsr.py | 78 ++++
.../{test_gemm.py => test_transposerowsp.py} | 69 ++--
.../{gemm_lib.cc => transposecsr_lib.cc} | 183 ++++-----
.../{gemm_lib.cc => transposerowsp_lib.cc} | 181 ++++-----
example/extensions/lib_subgraph/README.md | 69 ++--
example/extensions/lib_subgraph/subgraph_lib.cc | 45 ++-
example/extensions/lib_subgraph/test_subgraph.py | 62 ++-
include/mxnet/c_api.h | 4 +-
include/mxnet/lib_api.h | 422 +++++++++++++++++++--
include/mxnet/random_generator.h | 8 +
perl-package/AI-MXNetCAPI/mxnet.i | 2 +
python/mxnet/gluon/block.py | 71 +++-
python/mxnet/symbol/symbol.py | 23 +-
src/c_api/c_api.cc | 213 ++++++++---
src/c_api/c_api_symbolic.cc | 70 +++-
src/common/random_generator.cu | 5 +
src/operator/subgraph/build_subgraph.cc | 9 +-
.../partitioner/custom_subgraph_property.h | 247 ++++++++++--
src/operator/subgraph/subgraph_property.h | 8 +
tests/python/gpu/test_extensions_gpu.py | 18 +-
tests/python/unittest/test_extensions.py | 14 +
tests/python/unittest/test_subgraph_op.py | 4 +-
26 files changed, 1486 insertions(+), 477 deletions(-)
create mode 100644 example/extensions/lib_custom_op/test_transposecsr.py
copy example/extensions/lib_custom_op/{test_gemm.py => test_transposerowsp.py} (52%)
copy example/extensions/lib_custom_op/{gemm_lib.cc => transposecsr_lib.cc} (51%)
copy example/extensions/lib_custom_op/{gemm_lib.cc => transposerowsp_lib.cc} (51%)