@@ -91,25 +91,25 @@ typedef sycl::event (*nan_to_num_fn_ptr_t)(sycl::queue &,
91
91
const std::vector<sycl::event> &);
92
92
93
93
template <typename T>
94
- sycl::event nan_to_num_call (sycl::queue &exec_q,
95
- int nd,
96
- std::size_t nelems,
97
- const py::ssize_t *shape_strides,
98
- const py::object &py_nan,
99
- const py::object &py_posinf,
100
- const py::object &py_neginf,
101
- const char *arg_p,
102
- py::ssize_t arg_offset,
103
- char *dst_p,
104
- py::ssize_t dst_offset,
105
- const std::vector<sycl::event> &depends)
94
+ sycl::event nan_to_num_strided_call (sycl::queue &exec_q,
95
+ int nd,
96
+ std::size_t nelems,
97
+ const py::ssize_t *shape_strides,
98
+ const py::object &py_nan,
99
+ const py::object &py_posinf,
100
+ const py::object &py_neginf,
101
+ const char *arg_p,
102
+ py::ssize_t arg_offset,
103
+ char *dst_p,
104
+ py::ssize_t dst_offset,
105
+ const std::vector<sycl::event> &depends)
106
106
{
107
107
using dpctl::tensor::type_utils::is_complex_v;
108
108
using scT = std::conditional_t <is_complex_v<T>, value_type_of_t <T>, T>;
109
109
110
- scT nan_v = py::cast<scT>(py_nan);
111
- scT posinf_v = py::cast<scT>(py_posinf);
112
- scT neginf_v = py::cast<scT>(py_neginf);
110
+ const scT nan_v = py::cast<const scT>(py_nan);
111
+ const scT posinf_v = py::cast<const scT>(py_posinf);
112
+ const scT neginf_v = py::cast<const scT>(py_neginf);
113
113
114
114
using dpnp::kernels::nan_to_num::nan_to_num_impl;
115
115
sycl::event to_num_ev = nan_to_num_impl<T, scT>(
@@ -142,9 +142,9 @@ sycl::event nan_to_num_contig_call(sycl::queue &exec_q,
142
142
using dpctl::tensor::type_utils::is_complex_v;
143
143
using scT = std::conditional_t <is_complex_v<T>, value_type_of_t <T>, T>;
144
144
145
- scT nan_v = py::cast<scT>(py_nan);
146
- scT posinf_v = py::cast<scT>(py_posinf);
147
- scT neginf_v = py::cast<scT>(py_neginf);
145
+ const scT nan_v = py::cast<const scT>(py_nan);
146
+ const scT posinf_v = py::cast<const scT>(py_posinf);
147
+ const scT neginf_v = py::cast<const scT>(py_neginf);
148
148
149
149
using dpnp::kernels::nan_to_num::nan_to_num_contig_impl;
150
150
sycl::event to_num_contig_ev = nan_to_num_contig_impl<T, scT>(
@@ -331,7 +331,7 @@ std::pair<sycl::event, sycl::event>
331
331
332
332
/* *
333
333
* @brief A factory to define pairs of supported types for which
334
- * nan_to_num_call<T> function is available.
334
+ * nan-to-num function is available.
335
335
*
336
336
* @tparam T Type of input vector `a` and of result vector `y`.
337
337
*/
@@ -357,7 +357,7 @@ struct NanToNumFactory
357
357
return nullptr ;
358
358
}
359
359
else {
360
- return nan_to_num_call <T>;
360
+ return nan_to_num_strided_call <T>;
361
361
}
362
362
}
363
363
};
0 commit comments