replicate/megablocks

1#pragma once2 3#include <torch/all.h>4 5namespace megablocks {6 7// Public interface function for radix sorting with indices8void sort(torch::Tensor x,9          int end_bit,10          torch::Tensor x_out,11          torch::Tensor iota_out);12 13} // namespace megablocks