@@ -91,25 +91,25 @@ typedef sycl::event (*nan_to_num_fn_ptr_t)(sycl::queue &,
9191 const std::vector<sycl::event> &);
9292
9393template <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)
106106{
107107 using dpctl::tensor::type_utils::is_complex_v;
108108 using scT = std::conditional_t <is_complex_v<T>, value_type_of_t <T>, T>;
109109
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);
113113
114114 using dpnp::kernels::nan_to_num::nan_to_num_impl;
115115 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,
142142 using dpctl::tensor::type_utils::is_complex_v;
143143 using scT = std::conditional_t <is_complex_v<T>, value_type_of_t <T>, T>;
144144
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);
148148
149149 using dpnp::kernels::nan_to_num::nan_to_num_contig_impl;
150150 sycl::event to_num_contig_ev = nan_to_num_contig_impl<T, scT>(
@@ -331,7 +331,7 @@ std::pair<sycl::event, sycl::event>
331331
332332/* *
333333 * @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.
335335 *
336336 * @tparam T Type of input vector `a` and of result vector `y`.
337337 */
@@ -357,7 +357,7 @@ struct NanToNumFactory
357357 return nullptr ;
358358 }
359359 else {
360- return nan_to_num_call <T>;
360+ return nan_to_num_strided_call <T>;
361361 }
362362 }
363363};
0 commit comments