ColossalAI/docs/colossalai/colossalai.kernel.jit.rst

14 lines
237 B
ReStructuredText
Raw Normal View History

colossalai.kernel.jit
=====================
.. automodule:: colossalai.kernel.jit
:members:
.. toctree::
:maxdepth: 2
colossalai.kernel.jit.bias_dropout_add
colossalai.kernel.jit.bias_gelu
colossalai.kernel.jit.option