code: def func(self, runs):
      p = psutil.Process()
      # warm up for 5 runs, then things should be stable for the last 5
      last_rss = collections.deque(maxlen=5)
      for n in range(10):
          for i in range(runs):
              self.model(self.x)
          last_rss.append(p.memory_info().rss)
      return last_rss
  
file_path: pytorch/test/test_openmp.py
imports: import collections
  import unittest
  import torch
  from torch.testing._internal.common_utils import TestCase, run_tests, TEST_WITH_ASAN
label_desc: Use image node_id 1 for calling the TestOpenMP_ParallelFor obj's underlying member method code with example usage: obj.func(runs) and returns: last_rss