Enable _linalg_det, _linalg_slogdet and _linalg_solve_ex related operations on xpu

src/ATen/native/xpu/BatchLinearAlgebra.cpp

@@ -0,0 +1,52 @@
+#include <ATen/core/Tensor.h>
+#include <ATen/native/BatchLinearAlgebra.h>
+#include <ATen/native/DispatchStub.h>
+#if defined(USE_ONEMKL_XPU)
+#include <ATen/native/xpu/mkl/BatchLinearAlgebra.h>
+#endif // USE_ONEMKL_XPU
+
+namespace at::native {
+
+void lu_solve_kernel_xpu(
+    const Tensor& LU,
+    const Tensor& pivots,
+    const Tensor& B,
+    TransposeType trans) {
+#if defined(USE_ONEMKL_XPU)
+  native::xpu::lu_solve_mkl(LU, pivots, B, trans);
+#else
+  const auto LU_cpu = LU.to(LU.options().device(kCPU));
+  const auto pivots_cpu = pivots.to(pivots.options().device(kCPU));
+  auto B_cpu = B.to(B.options().device(kCPU));
+
+  lu_solve_stub(at::kCPU, LU_cpu, pivots_cpu, B_cpu, trans);
+
+  B.copy_(B_cpu);
+#endif // USE_ONEMKL_XPU
+}
+
+REGISTER_XPU_DISPATCH(lu_solve_stub, &lu_solve_kernel_xpu);
+
+void lu_factor_kernel_xpu(
+    const Tensor& input,
+    const Tensor& pivots,
+    const Tensor& infos,
+    bool compute_pivots) {
+#if defined(USE_ONEMKL_XPU)
+  native::xpu::lu_factor_mkl(input, pivots, infos, compute_pivots);
+#else
+  auto input_cpu = input.to(input.options().device(kCPU));
+  auto pivots_cpu = pivots.to(pivots.options().device(kCPU));
+  const auto infos_cpu = infos.to(infos.options().device(kCPU));
+
+  lu_factor_stub(at::kCPU, input_cpu, pivots_cpu, infos_cpu, compute_pivots);
+
+  input.copy_(input_cpu);
+  pivots.copy_(pivots_cpu);
+  infos.copy_(infos_cpu);
+#endif // USE_ONEMKL_XPU
+}
+
+REGISTER_XPU_DISPATCH(lu_factor_stub, &lu_factor_kernel_xpu);
+
+} // namespace at::native

src/ATen/native/xpu/XPUFallback.template

@@ -193,7 +193,6 @@ TORCH_LIBRARY_IMPL(aten, XPU, m) {
     "_flash_attention_forward",
     "geqrf",
     "linalg_cholesky_ex.L",
-    "_linalg_det.result",
     "linalg_eig",
     "_linalg_eigvals",
     "linalg_eigvals.out",
@@ -203,13 +202,9 @@ TORCH_LIBRARY_IMPL(aten, XPU, m) {
     "linalg_ldl_factor_ex.out",
     "linalg_ldl_solve.out",
     "linalg_lstsq.out",
-    "linalg_lu_factor_ex.out",
     "linalg_lu.out",
-    "linalg_lu_solve.out",
     "linalg_matrix_exp",
     "linalg_qr.out",
-    "_linalg_slogdet.sign",
-    "_linalg_solve_ex.result",
     "linalg_solve_triangular",
     "_linalg_svd.U",
     "lu_unpack.out",