code: def forward(
      ctx,
      b,
      row_indices,
      values,
      row_offsets,
      column_indices,
      m,
      _transp_info,
  ):
      b = b.contiguous()
      out = torch.ops.xformers.spmm_sputnik(
          b, row_indices, values, row_offsets, column_indices, m
      )
  
      ctx.save_for_backward(
          b,
          row_indices,
          values,
          row_offsets,
          column_indices,
          *_transp_info,
      )
      return out
  
file_path: xformers/xformers/sparse/_csr_ops.py
imports: import torch
  from .utils import _csr_to_coo, _transpose_with_info
label_desc: Use image node_id 1 for calling the _spmm obj's underlying member method code with example usage: obj.forward(ctx, b, row_indices, values, row_offsets, column_indices, m, _transp_info) and returns: out