fairseq-a54021305d6b3c4c5959ac9395135f63202db8f1/fairseq/modules/dynamicconv_layer/dynamiconv_cpu.cpp

#include <torch/torch.h>
#include <vector>

std::vector<float*>
dynamicconv_cpu_forward(float* input, float* filters, int padding_l);

std::vector<float*> dynamicconv_cpu_backward(
    float* gradOutput,
    int padding_l,
    float* input,
    float* filters);

std::vector<float*>
dynamicconv_forward(float* input, float* filters, int padding_l) {
  return dynamicconv_cpu_forward(input, filters, padding_l);
}

std::vector<float*> dynamicconv_backward(
    float* gradOutput,
    int padding_l,
    float* input,
    float* filters) {
  return dynamicconv_cpu_backward(gradOutput, padding_l, input, filters);
}

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("forward", &dynamicconv_forward, "dynamicconv forward (CPU)");
  m.def("backward", &dynamicconv_backward, "dynamicconv backward (CPU)");
}