mirror of https://github.com/hpcaitech/ColossalAI
[doc] update apidoc (#530)
parent
05e33b2578
commit
ffca99d187
|
@ -1,5 +0,0 @@
|
|||
colossalai.context.process\_group\_initializer.initializer\_moe
|
||||
===============================================================
|
||||
|
||||
.. automodule:: colossalai.context.process_group_initializer.initializer_moe
|
||||
:members:
|
|
@ -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
|
||||
|
|
|
@ -18,3 +18,4 @@ colossalai.context
|
|||
colossalai.context.moe_context
|
||||
colossalai.context.parallel_context
|
||||
colossalai.context.parallel_mode
|
||||
colossalai.context.singleton_meta
|
||||
|
|
|
@ -0,0 +1,5 @@
|
|||
colossalai.context.singleton\_meta
|
||||
==================================
|
||||
|
||||
.. automodule:: colossalai.context.singleton_meta
|
||||
:members:
|
|
@ -0,0 +1,5 @@
|
|||
colossalai.logging.logger
|
||||
=========================
|
||||
|
||||
.. automodule:: colossalai.logging.logger
|
||||
:members:
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.logging.logging
|
||||
==========================
|
||||
|
||||
.. automodule:: colossalai.logging.logging
|
||||
:members:
|
|
@ -8,4 +8,4 @@ colossalai.logging
|
|||
.. toctree::
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.logging.logging
|
||||
colossalai.logging.logger
|
||||
|
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.commons.bucket\_tensor\_copy
|
||||
=============================================
|
||||
|
||||
.. automodule:: colossalai.utils.commons.bucket_tensor_copy
|
||||
:members:
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.commons.memory
|
||||
===============================
|
||||
|
||||
.. automodule:: colossalai.utils.commons.memory
|
||||
:members:
|
|
@ -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
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.commons.singleton\_meta
|
||||
========================================
|
||||
|
||||
.. automodule:: colossalai.utils.commons.singleton_meta
|
||||
:members:
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.memory
|
||||
=======================
|
||||
|
||||
.. automodule:: colossalai.utils.memory
|
||||
:members:
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.memory\_tracer.allocator
|
||||
=========================================
|
||||
|
||||
.. automodule:: colossalai.utils.memory_tracer.allocator
|
||||
:members:
|
|
@ -1,5 +0,0 @@
|
|||
colossalai.utils.memory\_tracer.commons
|
||||
=======================================
|
||||
|
||||
.. automodule:: colossalai.utils.memory_tracer.commons
|
||||
:members:
|
|
@ -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
|
||||
|
|
|
@ -0,0 +1,5 @@
|
|||
colossalai.utils.memory\_utils.bucket\_tensor\_copy
|
||||
===================================================
|
||||
|
||||
.. automodule:: colossalai.utils.memory_utils.bucket_tensor_copy
|
||||
:members:
|
|
@ -0,0 +1,5 @@
|
|||
colossalai.utils.memory\_utils.memory\_monitor
|
||||
==============================================
|
||||
|
||||
.. automodule:: colossalai.utils.memory_utils.memory_monitor
|
||||
:members:
|
|
@ -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
|
|
@ -0,0 +1,5 @@
|
|||
colossalai.utils.memory\_utils.utils
|
||||
====================================
|
||||
|
||||
.. automodule:: colossalai.utils.memory_utils.utils
|
||||
:members:
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue