Add aten::_ctc_loss and its variants

src/ATen/native/xpu/LossCTC.cpp

@@ -0,0 +1,43 @@
+#include <ATen/core/Reduction.h>
+#include <ATen/core/Tensor.h>
+#include <ATen/native/xpu/sycl/LossCTCKernels.h>
+#include <comm/RegisterUtils.h>
+
+namespace at {
+namespace native {
+
+std::tuple<Tensor, Tensor> ctc_loss_xpu(
+    const Tensor& log_probs,
+    const Tensor& targets,
+    IntArrayRef input_lengths,
+    IntArrayRef target_lengths,
+    int64_t blank,
+    bool zero_infinity) {
+  return native::xpu::ctc_loss_kernel(
+      log_probs, targets, input_lengths, target_lengths, blank, zero_infinity);
+}
+
+Tensor ctc_loss_backward_xpu(
+    const Tensor& grad,
+    const Tensor& log_probs,
+    const Tensor& targets,
+    IntArrayRef input_lengths,
+    IntArrayRef target_lengths,
+    const Tensor& neg_log_likelihood,
+    const Tensor& log_alpha,
+    int64_t blank,
+    bool zero_infinity) {
+  return native::xpu::ctc_loss_backward_kernel(
+      grad,
+      log_probs,
+      targets,
+      input_lengths,
+      target_lengths,
+      neg_log_likelihood,
+      log_alpha,
+      blank,
+      zero_infinity);
+}
+
+} // namespace native
+} // namespace at

src/ATen/native/xpu/XPUFallback.template

@@ -166,8 +166,6 @@ TORCH_LIBRARY_IMPL(aten, XPU, m) {
     "cholesky",
     "cholesky_inverse",
     "_cholesky_solve_helper",
-    "_ctc_loss",
-    "_ctc_loss_backward",
     "_cummax_helper",
     "_cummin_helper",
     "dot",