class_docstr: Estimate memory with chunk
class_docstr_tok: ["Estimate","memory","with","chunk"]
code: def _print_compute_op_mem_log(self, log, nodes, title=None):
      if title:
          print(title)
      for idx, (l, n) in enumerate(zip(log, nodes)):
          if n.op in ["placeholder", "get_attr", "output"]:
              continue
          if any(i in n.name for i in ["getitem", "getattr"]):
              continue
          print("%s:%.2f \t" % (n.name, l), end="")
          if (idx + 1) % 3 == 0:
              print("")
      print("\n")
  
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 8 for calling the EstimateMemory obj's underlying member method code with example usage: obj._print_compute_op_mem_log(log, nodes, title) without return types