class_docstr: Manage token block indexes and allocate physical memory for key and value cache
  
  Args:
      size: maximum token number used as the size of key and value buffer
      dtype: data type of cached key and value
      head_num: number of heads the memory manager is responsible for
      head_dim: embedded size per head
      layer_num: the number of layers in the model
      device: device used to store the key and value cache
docstr: free all memory by updating memory states
code: def free_all(self):
      """free all memory by updating memory states"""
      self.available_size = len(self.mem_state)
      self.mem_state[:] = 1
      self.max_len_in_batch = 0
  
file_path: ColossalAI/colossalai/inference/kv_cache/kvcache_manager.py
imports: import torch
  from transformers.utils import logging
label_desc: Use image node_id 7 for calling the MemoryManager obj's underlying member method code with example usage: obj.free_all() without return types