class_docstr: Stateful Tensor Manager, inspired from PatrickStar
  
  PatrickStar: Parallel Training of Pre-trained Models via Chunk-based Memory Management
  https://arxiv.org/abs/2108.05818
class_docstr_tok: ["Stateful","Tensor","Manager",",","inspired","from","PatrickStar","PatrickStar",":","Parallel","Training","of","Pre-trained","Models","via","Chunk-based","Memory","Management","https",":","\/\/arxiv.org\/abs\/2108.05818"]
code: def cpu_gpu_move_volume(self):
      return self._cpu_gpu_move_volume
  
file_path: ColossalAI/colossalai/legacy/zero/gemini/stateful_tensor_mgr.py
imports: import functools
  import types
  from time import time
  from typing import List
  from colossalai.utils.device import get_current_device
  from .stateful_tensor import StatefulTensor, TensorState
  from .tensor_placement_policy import TensorPlacementPolicy
  from .tensor_utils import colo_model_data_tensor_move_inline
label_desc: Use image node_id 6 for calling the StatefulTensorMgr obj's underlying member method code with example usage: obj.cpu_gpu_move_volume() and returns: self