class_docstr: Estimate memory with chunk
class_docstr_tok: ["Estimate","memory","with","chunk"]
code: def _get_chunk_ratio(self, node, chunk_node_dim, chunk_size):
      if node not in chunk_node_dim:
          return 1.0
      node_shape = get_node_shape(node)
      chunk_dim = chunk_node_dim[node]["chunk_dim"]
      if chunk_dim is None:
          return 1.0
      else:
          return chunk_size / float(node_shape[chunk_dim])
  
file_path: ColossalAI/colossalai/autochunk/estimate_memory.py
imports: from typing import Dict, List
  import torch
  from torch.fx.node import Node
  from .utils import NodeMgr, get_node_shape, is_non_memory_node
label_desc: Use image node_id 7 for calling the EstimateMemory obj's underlying member method code with example usage: obj._get_chunk_ratio(node, chunk_node_dim, chunk_size) and returns: int, int, unknown