colossalai.utils.profiler.stateful\_tensor\_mem\_extention
==========================================================

.. automodule:: colossalai.utils.profiler.stateful_tensor_mem_extention
   :members: