Default int type queries behave based on NumPy version

Introduced default_device_uint_type, parallel to
default_device_int_type.

For NumPy >= 2 (as checked at runtime), it returns "i8" (since dpctl
is only supported on x86_64) or unsigned "u8", while for NumPy < 2 it
returns long ("l"), or 'unsigned long' ("L").

Author: oleksandr-pavlyk

dpctl/tensor/libtensor/source/device_support_queries.cpp

@@ -49,11 +49,48 @@ std::string _default_device_fp_type(const sycl::device &d)
     }
 }
 
+int get_numpy_major_version()
+{
+    namespace py = pybind11;
+
+    py::module_ numpy = py::module_::import("numpy");
+    py::str version_string = numpy.attr("__version__");
+    py::module_ numpy_lib = py::module_::import("numpy.lib");
+
+    py::object numpy_version = numpy_lib.attr("NumpyVersion")(version_string);
+    int major_version = numpy_version.attr("major").cast<int>();
+
+    return major_version;
+}
+
 std::string _default_device_int_type(const sycl::device &)
 {
-    return "l"; // code for numpy.dtype('long') to be consistent
-                // with NumPy's default integer type across
-                // platforms.
+    const int np_ver = get_numpy_major_version();
+
+    if (np_ver >= 2) {
+        return "i8";
+    }
+    else {
+        // code for numpy.dtype('long') to be consistent
+        // with NumPy's default integer type across
+        // platforms.
+        return "l";
+    }
+}
+
+std::string _default_device_uint_type(const sycl::device &)
+{
+    const int np_ver = get_numpy_major_version();
+
+    if (np_ver >= 2) {
+        return "u8";
+    }
+    else {
+        // code for numpy.dtype('long') to be consistent
+        // with NumPy's default integer type across
+        // platforms.
+        return "L";
+    }
 }
 
 std::string _default_device_complex_type(const sycl::device &d)
@@ -66,15 +103,9 @@ std::string _default_device_complex_type(const sycl::device &d)
     }
 }
 
-std::string _default_device_bool_type(const sycl::device &)
-{
-    return "b1";
-}
+std::string _default_device_bool_type(const sycl::device &) { return "b1"; }
 
-std::string _default_device_index_type(const sycl::device &)
-{
-    return "i8";
-}
+std::string _default_device_index_type(const sycl::device &) { return "i8"; }
 
 sycl::device _extract_device(const py::object &arg)
 {
@@ -108,6 +139,12 @@ std::string default_device_int_type(const py::object &arg)
     return _default_device_int_type(d);
 }
 
+std::string default_device_uint_type(const py::object &arg)
+{
+    const sycl::device &d = _extract_device(arg);
+    return _default_device_uint_type(d);
+}
+
 std::string default_device_bool_type(const py::object &arg)
 {
     const sycl::device &d = _extract_device(arg);

dpctl/tensor/libtensor/source/device_support_queries.hpp

@@ -39,6 +39,7 @@ namespace py_internal
 
 extern std::string default_device_fp_type(const py::object &);
 extern std::string default_device_int_type(const py::object &);
+extern std::string default_device_uint_type(const py::object &);
 extern std::string default_device_bool_type(const py::object &);
 extern std::string default_device_complex_type(const py::object &);
 extern std::string default_device_index_type(const py::object &);

dpctl/tensor/libtensor/source/tensor_ctors.cpp

@@ -331,7 +331,13 @@ PYBIND11_MODULE(_tensor_impl, m)
 
     m.def("default_device_int_type",
           dpctl::tensor::py_internal::default_device_int_type,
-          "Gives default integer type supported by device.", py::arg("dev"));
+          "Gives default signed integer type supported by device.",
+          py::arg("dev"));
+
+    m.def("default_device_uint_type",
+          dpctl::tensor::py_internal::default_device_uint_type,
+          "Gives default unsigned integer type supported by device.",
+          py::arg("dev"));
 
     m.def("default_device_bool_type",
           dpctl::tensor::py_internal::default_device_bool_type,