mirror/trigdx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use const size_t for n

Browse Source
This commit is contained in:
Bram Veenboer
2025-08-14 11:07:45 +02:00
parent 79dc7b4285
commit b3467840f9
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/bindings.cpp
View File

@@ -10,7 +10,7 @@ template <typename T>
py::array_t<T> py::array_t<T>
compute_sin(const Backend &backend, compute_sin(const Backend &backend,
py::array_t<T, py::array::c_style | py::array::forcecast> x) { py::array_t<T, py::array::c_style | py::array::forcecast> x) {
ssize_t n = x.shape(0); const size_t n = x.shape(0);
if (x.ndim() != 1) { if (x.ndim() != 1) {
throw py::value_error("Input array must be 1-dimensional"); throw py::value_error("Input array must be 1-dimensional");
} }
@@ -20,7 +20,7 @@ compute_sin(const Backend &backend,
py::array_t<float> s(n); py::array_t<float> s(n);
T *s_ptr = s.mutable_data(); T *s_ptr = s.mutable_data();
backend.compute_sinf(static_cast<size_t>(n), x_ptr, s_ptr); backend.compute_sinf(n, x_ptr, s_ptr);
return s; return s;
} }
@@ -29,7 +29,7 @@ template <typename T>
py::array_t<T> py::array_t<T>
compute_cos(const Backend &backend, compute_cos(const Backend &backend,
py::array_t<T, py::array::c_style | py::array::forcecast> x) { py::array_t<T, py::array::c_style | py::array::forcecast> x) {
ssize_t n = x.shape(0); const size_t n = x.shape(0);
if (x.ndim() != 1) { if (x.ndim() != 1) {
throw py::value_error("Input array must be 1-dimensional"); throw py::value_error("Input array must be 1-dimensional");
} }
@@ -39,7 +39,7 @@ compute_cos(const Backend &backend,
py::array_t<T> c(n); py::array_t<T> c(n);
T *c_ptr = c.mutable_data(); T *c_ptr = c.mutable_data();
backend.compute_cosf(static_cast<size_t>(n), x_ptr, c_ptr); backend.compute_cosf(n, x_ptr, c_ptr);
return c; return c;
} }
@@ -48,7 +48,7 @@ template <typename T>
std::tuple<py::array_t<T>, py::array_t<T>> std::tuple<py::array_t<T>, py::array_t<T>>
compute_sincos(const Backend &backend, compute_sincos(const Backend &backend,
py::array_t<T, py::array::c_style | py::array::forcecast> x) { py::array_t<T, py::array::c_style | py::array::forcecast> x) {
ssize_t n = x.shape(0); const size_t n = x.shape(0);
if (x.ndim() != 1) { if (x.ndim() != 1) {
throw py::value_error("Input array must be 1-dimensional"); throw py::value_error("Input array must be 1-dimensional");
} }
@@ -58,8 +58,7 @@ compute_sincos(const Backend &backend,
py::array_t<T> s(n); py::array_t<T> s(n);
py::array_t<T> c(n); py::array_t<T> c(n);
backend.compute_sincosf(static_cast<size_t>(n), x_ptr, s.mutable_data(), backend.compute_sincosf(n, x_ptr, s.mutable_data(), c.mutable_data());
c.mutable_data());
return std::make_tuple(s, c); return std::make_tuple(s, c);
} }