mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-28 04:55:25 +00:00
[doc] update rst and docstring (#1351)
* update rst * add zero docstr * fix docstr * remove fx.tracer.meta_patch * fix docstr * fix docstr * update fx rst * fix fx docstr * remove useless rst
This commit is contained in:
27
docs/colossalai/colossalai.gemini.rst
Normal file
27
docs/colossalai/colossalai.gemini.rst
Normal file
@@ -0,0 +1,27 @@
|
||||
colossalai.gemini
|
||||
=================
|
||||
|
||||
.. automodule:: colossalai.gemini
|
||||
:members:
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.gemini.memory_tracer
|
||||
colossalai.gemini.ophooks
|
||||
colossalai.gemini.paramhooks
|
||||
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.gemini.chunk
|
||||
colossalai.gemini.chunk_mgr
|
||||
colossalai.gemini.gemini_context
|
||||
colossalai.gemini.gemini_mgr
|
||||
colossalai.gemini.placement_policy
|
||||
colossalai.gemini.stateful_tensor
|
||||
colossalai.gemini.stateful_tensor_container
|
||||
colossalai.gemini.stateful_tensor_mgr
|
||||
colossalai.gemini.tensor_placement_policy
|
||||
colossalai.gemini.tensor_utils
|
Reference in New Issue
Block a user