#pragma once

#include <torch/extension.h>

#include <vector>
#include <tuple>

std::vector<torch::Tensor> dispatch_cuda(torch::Tensor input,
  int dim,
  bool descending,
  torch::optional<std::tuple<torch::Tensor, torch::Tensor>> out);