Skip to content

Commit 99fd28b

Browse files
committed
Apply review comments
* nan_to_num_call -> nan_to_num_strided_call * add missing const markers on converted Python scalar objects
1 parent e9953f5 commit 99fd28b

File tree

1 file changed

+20
-20
lines changed

1 file changed

+20
-20
lines changed

dpnp/backend/extensions/ufunc/elementwise_functions/nan_to_num.cpp

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -91,25 +91,25 @@ typedef sycl::event (*nan_to_num_fn_ptr_t)(sycl::queue &,
9191
const std::vector<sycl::event> &);
9292

9393
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)
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

Comments
 (0)