topk_transform_pybind.cu

#include "topk_transform.h"
#include "rocm_ops.hpp"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    TOPK_TRANSFORM_PYBIND;
}