Run-Time Compilation API

.. currentmodule:: mxnet.rtc

Overview

The RTC package contains tools for compiling and running CUDA code from python frontend. The compiled kernels can be used stand-alone or combined with autograd.Function or operator.CustomOpProp to support differentiation.

.. autosummary::
    :nosignatures:

    mxnet.rtc

API Reference

.. automodule:: mxnet.rtc
    :members: