code: def check_data(output, cpu_data):
      (
          consumed_size,
          reclaimed_size,
          reclaimed_pct,
      ) = get_consumed_reclaimed_size(output)
  
      # compare with data
      final_consumed_size_stats = cpu_data.consumed / 2**20
      final_reclaimed_size_stats = cpu_data.reclaimed / 2**20
      # numbers are within 2% equal
      assert isclose(
          final_consumed_size_stats, consumed_size, rel_tol=0.02
      ), f"Consumed {final_consumed_size_stats} vs reclaimed {cpu.consumed} memory"
      assert isclose(
          final_reclaimed_size_stats, reclaimed_size, rel_tol=0.02
      ), f"Reclaimed {final_reclaimed_size_stats} vs reclaimed {cpu.reclaimed} memory"
  
imports: import numpy
  import torch
  import re, gc
  from math import isclose
  import inspect
label_desc: Use image node_id 14 for calling a global function with example usage: check_data(output, cpu_data) without return types