code: def get_configs_bwd():
      configs = []
      for num_stages in [0, 1, 2, 3, 4]:
          for block_m in [32, 64, 128]:
              for block_n in [16, 32, 64, 128]:
                  if block_n > block_m:
                      continue
                  for num_warps in [1, 2, 4, 8]:
                      if 32 * num_warps * 32 > block_m * block_n:
                          continue
                      configs.append(
                          triton.Config(
                              {"BLOCK_M": block_m, "BLOCK_N": block_n},
                              num_stages=num_stages,
                              num_warps=num_warps,
                              pre_hook=init_to_zero("DQ"),
                          )
                      )
      return configs
  
file_path: apex/apex/contrib/openfold_triton/_mha_kernel.py
imports: import triton
  import triton.language
label_desc: Use image node_id 5 for calling a global function with example usage: get_configs_bwd() and returns: configs