html refactor (#555)

docs/colossalai/colossalai.utils.data_sampler.base_sampler.rst

-colossalai.utils.data\_sampler.base\_sampler
-============================================
-
-.. automodule:: colossalai.utils.data_sampler.base_sampler
-   :members:

docs/colossalai/colossalai.utils.data_sampler.data_parallel_sampler.rst

-colossalai.utils.data\_sampler.data\_parallel\_sampler
-======================================================
-
-.. automodule:: colossalai.utils.data_sampler.data_parallel_sampler
-   :members:

docs/colossalai/colossalai.utils.data_sampler.rst

@@ -3,10 +3,3 @@ colossalai.utils.data\_sampler
 
 .. automodule:: colossalai.utils.data_sampler
    :members:
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.utils.data_sampler.base_sampler
-   colossalai.utils.data_sampler.data_parallel_sampler

docs/colossalai/colossalai.utils.memory_tracer.async_memtracer.rst

-colossalai.utils.memory\_tracer.async\_memtracer
-================================================
-
-.. automodule:: colossalai.utils.memory_tracer.async_memtracer
-   :members:

docs/colossalai/colossalai.utils.memory_tracer.memstats_collector.rst

-colossalai.utils.memory\_tracer.memstats\_collector
-===================================================
-
-.. automodule:: colossalai.utils.memory_tracer.memstats_collector
-   :members:

docs/colossalai/colossalai.utils.memory_tracer.model_data_memtracer.rst

-colossalai.utils.memory\_tracer.model\_data\_memtracer
-======================================================
-
-.. automodule:: colossalai.utils.memory_tracer.model_data_memtracer
-   :members:

docs/colossalai/colossalai.utils.memory_tracer.rst

 colossalai.utils.memory\_tracer
-===============================
+=======================
 
 .. automodule:: colossalai.utils.memory_tracer
    :members:
\ No newline at end of file
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.utils.memory_tracer.async_memtracer
-   colossalai.utils.memory_tracer.memstats_collector
-   colossalai.utils.memory_tracer.model_data_memtracer
-   colossalai.utils.memory_tracer.test_async_memtracer
-   colossalai.utils.memory_tracer.test_memstats_collector

docs/colossalai/colossalai.utils.memory_tracer.test_async_memtracer.rst

-colossalai.utils.memory\_tracer.test\_async\_memtracer
-======================================================
-
-.. automodule:: colossalai.utils.memory_tracer.test_async_memtracer
-   :members:

docs/colossalai/colossalai.utils.memory_tracer.test_memstats_collector.rst

-colossalai.utils.memory\_tracer.test\_memstats\_collector
-=========================================================
-
-.. automodule:: colossalai.utils.memory_tracer.test_memstats_collector
-   :members:

docs/colossalai/colossalai.utils.memory_utils.bucket_tensor_copy.rst

-colossalai.utils.memory\_utils.bucket\_tensor\_copy
-===================================================
-
-.. automodule:: colossalai.utils.memory_utils.bucket_tensor_copy
-   :members:

docs/colossalai/colossalai.utils.memory_utils.memory_monitor.rst

-colossalai.utils.memory\_utils.memory\_monitor
-==============================================
-
-.. automodule:: colossalai.utils.memory_utils.memory_monitor
-   :members:

docs/colossalai/colossalai.utils.memory_utils.rst

-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

docs/colossalai/colossalai.utils.memory_utils.utils.rst

-colossalai.utils.memory\_utils.utils
-====================================
-
-.. automodule:: colossalai.utils.memory_utils.utils
-   :members:

docs/colossalai/colossalai.utils.moe.rst

 colossalai.utils.moe
-====================
+=======================
 
 .. automodule:: colossalai.utils.moe
    :members:
\ No newline at end of file

docs/colossalai/colossalai.utils.profiler.comm_profiler.rst

-colossalai.utils.profiler.comm\_profiler
-========================================
-
-.. automodule:: colossalai.utils.profiler.comm_profiler
-   :members:

docs/colossalai/colossalai.utils.profiler.pcie_profiler.rst

-colossalai.utils.profiler.pcie\_profiler
-========================================
-
-.. automodule:: colossalai.utils.profiler.pcie_profiler
-   :members:

docs/colossalai/colossalai.utils.profiler.prof_utils.rst

-colossalai.utils.profiler.prof\_utils
-=====================================
-
-.. automodule:: colossalai.utils.profiler.prof_utils
-   :members:

docs/colossalai/colossalai.utils.profiler.rst

 colossalai.utils.profiler
-=========================
+=======================
 
 .. automodule:: colossalai.utils.profiler
    :members:
\ No newline at end of file
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.utils.profiler.comm_profiler
-   colossalai.utils.profiler.pcie_profiler
-   colossalai.utils.profiler.prof_utils

docs/colossalai/colossalai.utils.rst

 colossalai.utils
 ================
 
-.. automodule:: colossalai.utils
-   :members:
+*This part contains useful utils for colossalai*
 
 .. toctree::
    :maxdepth: 2
@@ -10,9 +9,8 @@ colossalai.utils
    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.profile
    colossalai.utils.tensor_detector
 
 
@@ -23,5 +21,6 @@ colossalai.utils
    colossalai.utils.checkpointing
    colossalai.utils.common
    colossalai.utils.cuda
+   colossalai.utils.memory
    colossalai.utils.moe
    colossalai.utils.timer

docs/colossalai/colossalai.utils.tensor_detector.rst

 colossalai.utils.tensor\_detector
-=================================
+=======================
 
-.. automodule:: colossalai.utils.tensor_detector
+.. automodule:: colossalai.utils.tensor_detector.tensor_detector
    :members:
\ No newline at end of file
-
-
-.. toctree::
-   :maxdepth: 2
-
-   colossalai.utils.tensor_detector.tensor_detector