Expose Transpose enum to Python via pybind11

Author: vlad-perevezentsev

dpnp/backend/extensions/lapack/getrs.cpp

@@ -166,7 +166,7 @@ std::pair<sycl::event, sycl::event>
           const dpctl::tensor::usm_ndarray &a_array,
           const dpctl::tensor::usm_ndarray &ipiv_array,
           const dpctl::tensor::usm_ndarray &b_array,
-          const int trans_code,
+          oneapi::mkl::transpose trans,
           const std::vector<sycl::event> &depends)
 {
     const int a_array_nd = a_array.get_ndim();
@@ -265,21 +265,6 @@ std::pair<sycl::event, sycl::event>
     const std::int64_t lda = std::max<size_t>(1UL, n);
     const std::int64_t ldb = std::max<size_t>(1UL, n);
 
-    oneapi::mkl::transpose trans;
-    switch (trans_code) {
-    case 0:
-        trans = oneapi::mkl::transpose::N;
-        break;
-    case 1:
-        trans = oneapi::mkl::transpose::T;
-        break;
-    case 2:
-        trans = oneapi::mkl::transpose::C;
-        break;
-    default:
-        throw py::value_error("`trans_code` must be 0 (N), 1 (T), or 2 (C)");
-    }
-
     char *a_array_data = a_array.get_data();
     char *b_array_data = b_array.get_data();
     char *ipiv_array_data = ipiv_array.get_data();

dpnp/backend/extensions/lapack/getrs.hpp

@@ -37,7 +37,7 @@ extern std::pair<sycl::event, sycl::event>
           const dpctl::tensor::usm_ndarray &a_array,
           const dpctl::tensor::usm_ndarray &ipiv_array,
           const dpctl::tensor::usm_ndarray &b_array,
-          const int trans_code,
+          oneapi::mkl::transpose trans,
           const std::vector<sycl::event> &depends = {});
 
 extern void init_getrs_dispatch_vector(void);

dpnp/backend/extensions/lapack/lapack_py.cpp

@@ -76,6 +76,13 @@ void init_dispatch_tables(void)
 
 PYBIND11_MODULE(_lapack_impl, m)
 {
+    // Expose oneMKL transpose enum to Python
+    py::enum_<oneapi::mkl::transpose>(m, "Transpose")
+        .value("N", oneapi::mkl::transpose::N)
+        .value("T", oneapi::mkl::transpose::T)
+        .value("C", oneapi::mkl::transpose::C)
+        .export_values(); // Optional, allows access like `Transpose.N`
+
     // Register a custom LinAlgError exception in the dpnp.linalg submodule
     py::module_ linalg_module = py::module_::import("dpnp.linalg");
     py::register_exception<lapack_ext::LinAlgError>(
@@ -160,7 +167,7 @@ PYBIND11_MODULE(_lapack_impl, m)
           "the solves of linear equations with an LU-factored "
           "square coefficient matrix, with multiple right-hand sides",
           py::arg("sycl_queue"), py::arg("a_array"), py::arg("ipiv_array"),
-          py::arg("b_array"), py::arg("trans_code"),
+          py::arg("b_array"), py::arg("trans") = oneapi::mkl::transpose::N,
           py::arg("depends") = py::list());
 
     m.def("_orgqr_batch", &lapack_ext::orgqr_batch,