code: def generate_matmul_test_data(dir, M, N, K):
      a = np.random.randn(M * K).astype(np.float16).reshape((M, K))
      b = np.random.randn(M * K).astype(np.float16).reshape((K, N))
      a_path = os.path.join(dir, "a.csv")
      b_path = os.path.join(dir, "b.csv")
      c_path = os.path.join(dir, "c.csv")
      for x, path in [(a, a_path), (b, b_path)]:
          x.view(np.int16).ravel().tofile(path, sep=",")
      return a, b, a_path, b_path, c_path
  
imports: import glob
  import os
  import subprocess
  import sys
  import tempfile
  import numpy
  import triton
  from triton.backends.nvidia.driver import include_dir, library_dir
label_desc: Use image node_id 8 for calling a global function with example usage: generate_matmul_test_data(dir, M, N, K) and returns: a,  b,  a_path,  b_path,  c_path