code: def test_matmul():
      for case in problem_list:
          M, N, K, TRANS_A, TRANS_B = case
          print(M, N, K, TRANS_A, TRANS_B)
          if TRANS_A:
              a = torch.randn(
                  (K, M), device="cuda", dtype=torch.float16
              ).T
              a_order = [0, 1]
          else:
              a = torch.randn(
                  (M, K), device="cuda", dtype=torch.float16
              )
              a_order = [1, 0]
  
          if TRANS_B:
              b = torch.randn(
                  (N, K), device="cuda", dtype=torch.float16
              ).T
              b_order = [0, 1]
          else:
              b = torch.randn(
                  (K, N), device="cuda", dtype=torch.float16
              )
              b_order = [1, 0]
  
          golden = torch.matmul(a, b)
          z = matmul(a, b, a_order, b_order)
  
          golden = torch.nn.functional.normalize(golden)
          z = torch.nn.functional.normalize(z)
          torch.set_printoptions(profile="full")
          assert_close(
              z, golden, rtol=1e-2, atol=1e-3, check_dtype=False
          )
  
imports: import torch
  from torch.testing import assert_close
  import triton
  import triton.language
label_desc: Use image node_id 3 for calling a global function with example usage: test_matmul() without return types