Untangled input type overloads for functions to get default type cate…

…gory for queue/device
IntelPython · May 16, 2022 · ea08c0e · ea08c0e
1 parent 2c3eea7
commit ea08c0e
Showing 1 changed file with 19 additions and 13 deletions.
diff --git a/dpctl/tensor/libtensor/source/tensor_py.cpp b/dpctl/tensor/libtensor/source/tensor_py.cpp
@@ -1946,28 +1946,34 @@ PYBIND11_MODULE(_tensor_impl, m)
           py::arg("fill_value"), py::arg("dst"), py::arg("sycl_queue"),
           py::arg("depends") = py::list());
 
-    m.def("default_device_fp_type", [](sycl::queue q) {
+    m.def("default_device_fp_type", [](sycl::queue q) -> std::string {
         return get_default_device_fp_type(q.get_device());
     });
-    m.def("default_device_fp_type",
-          [](sycl::device dev) { return get_default_device_fp_type(dev); });
+    m.def("default_device_fp_type_device", [](sycl::device dev) -> std::string {
+        return get_default_device_fp_type(dev);
+    });
 
-    m.def("default_device_int_type", [](sycl::queue q) {
+    m.def("default_device_int_type", [](sycl::queue q) -> std::string {
         return get_default_device_int_type(q.get_device());
     });
-    m.def("default_device_int_type",
-          [](sycl::device dev) { return get_default_device_int_type(dev); });
+    m.def("default_device_int_type_device",
+          [](sycl::device dev) -> std::string {
+              return get_default_device_int_type(dev);
+          });
 
-    m.def("default_device_bool_type", [](sycl::queue q) {
+    m.def("default_device_bool_type", [](sycl::queue q) -> std::string {
         return get_default_device_bool_type(q.get_device());
     });
-    m.def("default_device_bool_type",
-          [](sycl::device dev) { return get_default_device_bool_type(dev); });
+    m.def("default_device_bool_type_device",
+          [](sycl::device dev) -> std::string {
+              return get_default_device_bool_type(dev);
+          });
 
-    m.def("default_device_complex_type", [](sycl::queue q) {
+    m.def("default_device_complex_type", [](sycl::queue q) -> std::string {
         return get_default_device_complex_type(q.get_device());
     });
-    m.def("default_device_complex_type", [](sycl::device dev) {
-        return get_default_device_complex_type(dev);
-    });
+    m.def("default_device_complex_type_device",
+          [](sycl::device dev) -> std::string {
+              return get_default_device_complex_type(dev);
+          });
 }