Unverified Commit 9e3b8b7a authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[doc] removed read-the-docs (#2932)

parent 77b88a38
colossalai.utils
================
.. automodule:: colossalai.utils
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.checkpoint
colossalai.utils.data_sampler
colossalai.utils.model
colossalai.utils.multi_tensor_apply
colossalai.utils.profiler
colossalai.utils.tensor_detector
.. toctree::
:maxdepth: 2
colossalai.utils.activation_checkpoint
colossalai.utils.checkpointing
colossalai.utils.common
colossalai.utils.cuda
colossalai.utils.memory
colossalai.utils.moe
colossalai.utils.timer
colossalai.utils.tensor\_detector
=================================
.. automodule:: colossalai.utils.tensor_detector
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.tensor_detector.tensor_detector
colossalai.utils.tensor\_detector.tensor\_detector
==================================================
.. automodule:: colossalai.utils.tensor_detector.tensor_detector
:members:
colossalai.utils.timer
======================
.. automodule:: colossalai.utils.timer
:members:
colossalai.zero.init\_ctx.init\_context
=======================================
.. automodule:: colossalai.zero.init_ctx.init_context
:members:
colossalai.zero.init\_ctx
=========================
.. automodule:: colossalai.zero.init_ctx
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.init_ctx.init_context
colossalai.zero
===============
.. automodule:: colossalai.zero
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.init_ctx
colossalai.zero.shard_utils
colossalai.zero.sharded_model
colossalai.zero.sharded_optim
colossalai.zero.sharded_param
colossalai.zero.utils
.. toctree::
:maxdepth: 2
colossalai.zero.zero_optimizer
colossalai.zero.shard\_utils.base\_shard\_strategy
==================================================
.. automodule:: colossalai.zero.shard_utils.base_shard_strategy
:members:
colossalai.zero.shard\_utils.bucket\_tensor\_shard\_strategy
============================================================
.. automodule:: colossalai.zero.shard_utils.bucket_tensor_shard_strategy
:members:
colossalai.zero.shard\_utils.commons
====================================
.. automodule:: colossalai.zero.shard_utils.commons
:members:
colossalai.zero.shard\_utils
============================
.. automodule:: colossalai.zero.shard_utils
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.shard_utils.base_shard_strategy
colossalai.zero.shard_utils.bucket_tensor_shard_strategy
colossalai.zero.shard_utils.commons
colossalai.zero.shard_utils.tensor_shard_strategy
colossalai.zero.shard\_utils.tensor\_shard\_strategy
====================================================
.. automodule:: colossalai.zero.shard_utils.tensor_shard_strategy
:members:
colossalai.zero.sharded\_model.reduce\_scatter
==============================================
.. automodule:: colossalai.zero.sharded_model.reduce_scatter
:members:
colossalai.zero.sharded\_model
==============================
.. automodule:: colossalai.zero.sharded_model
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.sharded_model.reduce_scatter
colossalai.zero.sharded_model.sharded_model_v2
colossalai.zero.sharded_model.utils
colossalai.zero.sharded\_model.sharded\_model\_v2
=================================================
.. automodule:: colossalai.zero.sharded_model.sharded_model_v2
:members:
colossalai.zero.sharded\_model.utils
====================================
.. automodule:: colossalai.zero.sharded_model.utils
:members:
colossalai.zero.sharded\_optim
==============================
.. automodule:: colossalai.zero.sharded_optim
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.sharded_optim.sharded_optim_v2
colossalai.zero.sharded\_optim.sharded\_optim\_v2
=================================================
.. automodule:: colossalai.zero.sharded_optim.sharded_optim_v2
:members:
colossalai.zero.sharded\_param
==============================
.. automodule:: colossalai.zero.sharded_param
:members:
.. toctree::
:maxdepth: 2
colossalai.zero.sharded_param.sharded_param
colossalai.zero.sharded_param.sharded_tensor
colossalai.zero.sharded\_param.sharded\_param
=============================================
.. automodule:: colossalai.zero.sharded_param.sharded_param
: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