Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

 

New features

...

MXNet Extensions: custom operators, partitioning, and graph passes

Adds support for extending MXNet with custom operators, partitioning strategies, and graph passes. All implemented in a library easily compiled separately from the MXNet codebase, and dynamically loaded at runtime into any prebuilt installation of MXNet. 

  • fix for number of inputs/outputs for backward custom ops (#17069)
  • Enhancements for custom subgraph op (#17194)
  • Disable flaky test_custom_op_fork (#17481)
  • fix custom op makefile (#17516)
  • Update CustomOp doc with changes for GPU support (#17486)
  • [WIP] MXNet Extensions enhancements (#17885) (#18128)
  • Dynamic subgraph property (#17034)
  • Dynamic subgraph property doc (#17585)
  • [1.7] Backport MXNet Extension PRs (#17623, #17569, #17762) #18063 (#18069)

...