text: // Copyright (c) Facebook, Inc. and its affiliates.
  //
  // This source code is licensed under the MIT license found in the
  // LICENSE file in the root directory of this source tree.
  
  #include "../include/ball_query.h"
  #include "../include/group_points.h"
  #include "../include/interpolate.h"
  #include "../include/sampling.h"
  
  PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
    m.def("gather_points", &gather_points);
    m.def("gather_points_grad", &gather_points_grad);
    m.def("furthest_point_sampling", &furthest_point_sampling);
  
    m.def("three_nn", &three_nn);
    m.def("three_interpolate", &three_interpolate);
    m.def("three_interpolate_grad", &three_interpolate_grad);
  
    m.def("ball_query", &ball_query);
  
    m.def("group_points", &group_points);
    m.def("group_points_grad", &group_points_grad);
  }
  