
#include <torch/extension.h>

at::Tensor psa_forward_cpu(const torch::Tensor& hc, const int forward_type) {
  AT_ERROR("Not implemented on the CPU");
}

at::Tensor psa_backward_cpu(const at::Tensor& dout,
                            const at::Tensor& hc,
                            const int forward_type) {
  AT_ERROR("Not implemented on the CPU");
}