code: def find_di_locations(source: str) -> Dict[int, int]:
      """
      Parse out DILocation references in printed LLVM IR
      """
      result = {}
  
      for line in source.splitlines():
          m = re.match(r"!(\d+) = !DILocation\(line: (\d+).*", line)
          if m:
              debug_id, line = m.groups()
              result[debug_id] = line
  
      return result
  
file_path: tvm/tests/python/tir-base/test_debug_info.py
imports: import tvm
  import tvm.testing
  from tvm import tir
  from tvm import relay
  from tvm.script import tir
  from typing import List, Dict
  import re
label_desc: Use image node_id 1 for calling a global function with example usage: find_di_locations(source) and returns: result