[doc] update apidoc (#530)

pull/529/head^2
ver217 2022-03-25 18:29:43 +08:00 committed by GitHub
parent 05e33b2578
commit ffca99d187
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
20 changed files with 41 additions and 59 deletions

View File

@ -1,5 +0,0 @@
colossalai.context.process\_group\_initializer.initializer\_moe
===============================================================
.. automodule:: colossalai.context.process_group_initializer.initializer_moe
:members:

View File

@ -14,7 +14,6 @@ colossalai.context.process\_group\_initializer
colossalai.context.process_group_initializer.initializer_3d
colossalai.context.process_group_initializer.initializer_data
colossalai.context.process_group_initializer.initializer_model
colossalai.context.process_group_initializer.initializer_moe
colossalai.context.process_group_initializer.initializer_pipeline
colossalai.context.process_group_initializer.initializer_sequence
colossalai.context.process_group_initializer.initializer_tensor

View File

@ -18,3 +18,4 @@ colossalai.context
colossalai.context.moe_context
colossalai.context.parallel_context
colossalai.context.parallel_mode
colossalai.context.singleton_meta

View File

@ -0,0 +1,5 @@
colossalai.context.singleton\_meta
==================================
.. automodule:: colossalai.context.singleton_meta
:members:

View File

@ -0,0 +1,5 @@
colossalai.logging.logger
=========================
.. automodule:: colossalai.logging.logger
:members:

View File

@ -1,5 +0,0 @@
colossalai.logging.logging
==========================
.. automodule:: colossalai.logging.logging
:members:

View File

@ -8,4 +8,4 @@ colossalai.logging
.. toctree::
:maxdepth: 2
colossalai.logging.logging
colossalai.logging.logger

View File

@ -1,5 +0,0 @@
colossalai.utils.commons.bucket\_tensor\_copy
=============================================
.. automodule:: colossalai.utils.commons.bucket_tensor_copy
:members:

View File

@ -1,5 +0,0 @@
colossalai.utils.commons.memory
===============================
.. automodule:: colossalai.utils.commons.memory
:members:

View File

@ -1,13 +0,0 @@
colossalai.utils.commons
========================
.. automodule:: colossalai.utils.commons
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.commons.bucket_tensor_copy
colossalai.utils.commons.memory
colossalai.utils.commons.singleton_meta

View File

@ -1,5 +0,0 @@
colossalai.utils.commons.singleton\_meta
========================================
.. automodule:: colossalai.utils.commons.singleton_meta
:members:

View File

@ -1,5 +0,0 @@
colossalai.utils.memory
=======================
.. automodule:: colossalai.utils.memory
:members:

View File

@ -1,5 +0,0 @@
colossalai.utils.memory\_tracer.allocator
=========================================
.. automodule:: colossalai.utils.memory_tracer.allocator
:members:

View File

@ -1,5 +0,0 @@
colossalai.utils.memory\_tracer.commons
=======================================
.. automodule:: colossalai.utils.memory_tracer.commons
:members:

View File

@ -8,9 +8,7 @@ colossalai.utils.memory\_tracer
.. toctree::
:maxdepth: 2
colossalai.utils.memory_tracer.allocator
colossalai.utils.memory_tracer.async_memtracer
colossalai.utils.memory_tracer.commons
colossalai.utils.memory_tracer.memstats_collector
colossalai.utils.memory_tracer.model_data_memtracer
colossalai.utils.memory_tracer.test_async_memtracer

View File

@ -0,0 +1,5 @@
colossalai.utils.memory\_utils.bucket\_tensor\_copy
===================================================
.. automodule:: colossalai.utils.memory_utils.bucket_tensor_copy
:members:

View File

@ -0,0 +1,5 @@
colossalai.utils.memory\_utils.memory\_monitor
==============================================
.. automodule:: colossalai.utils.memory_utils.memory_monitor
:members:

View File

@ -0,0 +1,13 @@
colossalai.utils.memory\_utils
==============================
.. automodule:: colossalai.utils.memory_utils
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.memory_utils.bucket_tensor_copy
colossalai.utils.memory_utils.memory_monitor
colossalai.utils.memory_utils.utils

View File

@ -0,0 +1,5 @@
colossalai.utils.memory\_utils.utils
====================================
.. automodule:: colossalai.utils.memory_utils.utils
:members:

View File

@ -7,10 +7,10 @@ colossalai.utils
.. toctree::
:maxdepth: 2
colossalai.utils.commons
colossalai.utils.data_sampler
colossalai.utils.gradient_accumulation
colossalai.utils.memory_tracer
colossalai.utils.memory_utils
colossalai.utils.multi_tensor_apply
colossalai.utils.profiler
colossalai.utils.tensor_detector
@ -23,6 +23,5 @@ colossalai.utils
colossalai.utils.checkpointing
colossalai.utils.common
colossalai.utils.cuda
colossalai.utils.memory
colossalai.utils.moe
colossalai.utils.timer