code: def get_dense_mat_by_mask(val, mask):
      m, n_chunks, _ = mask.shape
      val = val.reshape(m, n_chunks, 2)
      ret = np.zeros((m, n_chunks, 4)).astype(val.dtype)
      for i in range(m):
          for j in range(n_chunks):
              for k in range(2):
                  ret[i, j, mask[i, j, k]] = val[i, j, k]
      return ret.reshape(m, n_chunks * 4)
  
file_path: tvm/tests/python/tir-base/test_tir_ptx_mma_sp.py
imports: import tvm
  from tvm.script import tir
  import numpy
  import tvm.testing
label_desc: Use image node_id 2 for calling a global function with example usage: get_dense_mat_by_mask(val, mask) and returns: ret