code: def _layer_norm_backward_buf_reduce(
      partial_buf_ptr,
      output_ptr,
      N: tl.constexpr,
      M: tl.constexpr,
      N_STRIDE: tl.constexpr,
      M_STRIDE: tl.constexpr,
  ):
      idx = N_STRIDE * tl.program_id(0) + M_STRIDE * tl.arange(0, M)
      mask = tl.program_id(0) < N
      x = tl.sum(
          tl.load(partial_buf_ptr + idx, mask, other=0).to(tl.float32),
          axis=0,
      )
      tl.store(output_ptr + tl.program_id(0), x, mask)
  
file_path: apex/apex/contrib/openfold_triton/_layer_norm_backward_kernels.py
imports: import torch
  import triton
  import triton.language
  from triton import Config
label_desc: Use image node_id 5 for calling a global function with example usage: _layer_norm_backward_buf_reduce(partial_buf_ptr, output_ptr, N, M, N_STRIDE, M_STRIDE) without return types