Skip to content

Commit c372cb7

Browse files
Untangled input type overloads for functions to get default type category for queue/device
1 parent 2a6314c commit c372cb7

File tree

1 file changed

+19
-13
lines changed

1 file changed

+19
-13
lines changed

dpctl/tensor/libtensor/source/tensor_py.cpp

Lines changed: 19 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1946,28 +1946,34 @@ PYBIND11_MODULE(_tensor_impl, m)
19461946
py::arg("fill_value"), py::arg("dst"), py::arg("sycl_queue"),
19471947
py::arg("depends") = py::list());
19481948

1949-
m.def("default_device_fp_type", [](sycl::queue q) {
1949+
m.def("default_device_fp_type", [](sycl::queue q) -> std::string {
19501950
return get_default_device_fp_type(q.get_device());
19511951
});
1952-
m.def("default_device_fp_type",
1953-
[](sycl::device dev) { return get_default_device_fp_type(dev); });
1952+
m.def("default_device_fp_type_device", [](sycl::device dev) -> std::string {
1953+
return get_default_device_fp_type(dev);
1954+
});
19541955

1955-
m.def("default_device_int_type", [](sycl::queue q) {
1956+
m.def("default_device_int_type", [](sycl::queue q) -> std::string {
19561957
return get_default_device_int_type(q.get_device());
19571958
});
1958-
m.def("default_device_int_type",
1959-
[](sycl::device dev) { return get_default_device_int_type(dev); });
1959+
m.def("default_device_int_type_device",
1960+
[](sycl::device dev) -> std::string {
1961+
return get_default_device_int_type(dev);
1962+
});
19601963

1961-
m.def("default_device_bool_type", [](sycl::queue q) {
1964+
m.def("default_device_bool_type", [](sycl::queue q) -> std::string {
19621965
return get_default_device_bool_type(q.get_device());
19631966
});
1964-
m.def("default_device_bool_type",
1965-
[](sycl::device dev) { return get_default_device_bool_type(dev); });
1967+
m.def("default_device_bool_type_device",
1968+
[](sycl::device dev) -> std::string {
1969+
return get_default_device_bool_type(dev);
1970+
});
19661971

1967-
m.def("default_device_complex_type", [](sycl::queue q) {
1972+
m.def("default_device_complex_type", [](sycl::queue q) -> std::string {
19681973
return get_default_device_complex_type(q.get_device());
19691974
});
1970-
m.def("default_device_complex_type", [](sycl::device dev) {
1971-
return get_default_device_complex_type(dev);
1972-
});
1975+
m.def("default_device_complex_type_device",
1976+
[](sycl::device dev) -> std::string {
1977+
return get_default_device_complex_type(dev);
1978+
});
19731979
}

0 commit comments

Comments
 (0)