{
  "small_batch": {
    "BLOCK_M": 32,
    "BLOCK_N": 64,
    "num_warps": 4,
    "num_stages": 2,
    "waves_per_eu": 0,
    "matrix_instr_nonkdim": 32,
    "kpack": 2,
    "SEQUENCE_PARALLEL": true,
    "UNROLL": 1
  },

  "large_batch": {
    "BLOCK_M": 32,
    "BLOCK_N": 64,
    "num_warps": 4,
    "num_stages": 1,
    "waves_per_eu": 0,
    "matrix_instr_nonkdim": 16,
    "kpack": 2,
    "SEQUENCE_PARALLEL": false,
    "UNROLL": 1
  }
}
