code: def test_128(self):
      self.run_test(128, 32, False)
      self.run_test(128, 32, True)
      self.run_test(128, 56, False)
      self.run_test(128, 56, True)
  
file_path: apex/apex/contrib/test/fmha/test_fmha.py
imports: import math
  import unittest
  import torch
  import numpy
label_desc: Use image node_id 2 for calling the TestFMHA obj's underlying member method code with example usage: obj.test_128() without return types