Commit be85a0f3 authored by Frank Lee's avatar Frank Lee Committed by ver217
Browse files

removed tutorial markdown and refreshed rst files for consistency

parent ca4ae52d
colossalai.context colossalai.context
================== ==================
.. automodule:: colossalai.context
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
...@@ -14,7 +17,3 @@ colossalai.context ...@@ -14,7 +17,3 @@ colossalai.context
colossalai.context.config colossalai.context.config
colossalai.context.parallel_context colossalai.context.parallel_context
colossalai.context.parallel_mode colossalai.context.parallel_mode
.. automodule:: colossalai.context
:members:
colossalai.core
===============
.. automodule:: colossalai.core
:members:
colossalai.engine colossalai.engine
================= =================
.. automodule:: colossalai.engine
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.engine.gradient_handler colossalai.engine.gradient_handler
colossalai.engine.schedule colossalai.engine.schedule
.. automodule:: colossalai.engine
:members:
colossalai.kernel.cuda\_native.layer\_norm
==========================================
.. automodule:: colossalai.kernel.cuda_native.layer_norm
:members:
colossalai.kernel.cuda\_native.multihead\_attention
===================================================
.. automodule:: colossalai.kernel.cuda_native.multihead_attention
:members:
colossalai.kernel.cuda\_native
==============================
.. automodule:: colossalai.kernel.cuda_native
:members:
.. toctree::
:maxdepth: 2
colossalai.kernel.cuda_native.layer_norm
colossalai.kernel.cuda_native.multihead_attention
colossalai.kernel.cuda_native.scaled_softmax
colossalai.kernel.cuda\_native.scaled\_softmax
==============================================
.. automodule:: colossalai.kernel.cuda_native.scaled_softmax
:members:
colossalai.kernel.jit.bias\_dropout\_add
========================================
.. automodule:: colossalai.kernel.jit.bias_dropout_add
:members:
colossalai.kernel.jit.bias\_gelu
================================
.. automodule:: colossalai.kernel.jit.bias_gelu
:members:
colossalai.kernel.jit.option
============================
.. automodule:: colossalai.kernel.jit.option
:members:
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
colossalai.kernel
=================
.. automodule:: colossalai.kernel
:members:
.. toctree::
:maxdepth: 2
colossalai.kernel.cuda_native
colossalai.kernel.jit
colossalai.logging colossalai.logging
================== ==================
.. automodule:: colossalai.logging
:members:
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.logging.logging colossalai.logging.logging
.. automodule:: colossalai.logging
:members:
colossalai.nn.init
==================
.. automodule:: colossalai.nn.init
:members:
colossalai.nn.layer.colossalai\_layer.dropout
=============================================
.. automodule:: colossalai.nn.layer.colossalai_layer.dropout
:members:
colossalai.nn.layer.colossalai\_layer.embedding
===============================================
.. automodule:: colossalai.nn.layer.colossalai_layer.embedding
:members:
colossalai.nn.layer.colossalai\_layer.linear
============================================
.. automodule:: colossalai.nn.layer.colossalai_layer.linear
:members:
colossalai.nn.layer.colossalai\_layer.normalization
===================================================
.. automodule:: colossalai.nn.layer.colossalai_layer.normalization
:members:
colossalai.nn.layer.colossalai\_layer
=====================================
.. automodule:: colossalai.nn.layer.colossalai_layer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.colossalai_layer.dropout
colossalai.nn.layer.colossalai_layer.embedding
colossalai.nn.layer.colossalai_layer.linear
colossalai.nn.layer.colossalai_layer.normalization
colossalai.nn.layer.moe.layers
==============================
.. automodule:: colossalai.nn.layer.moe.layers
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment