Skip to content

Commit 5b8c4e2

Browse files
committed
Add missing includes
1 parent e7bf84c commit 5b8c4e2

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

dpnp/backend/extensions/indexing/choose.cpp

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
#include <cstddef>
3131
#include <cstdint>
3232
#include <memory>
33+
#include <stdexcept>
34+
#include <string>
3335
#include <type_traits>
3436
#include <utility>
3537
#include <vector>
@@ -46,13 +48,16 @@
4648
// dpctl tensor headers
4749
#include "utils/indexing_utils.hpp"
4850
#include "utils/memory_overlap.hpp"
49-
#include "utils/offset_utils.hpp" //
51+
#include "utils/offset_utils.hpp"
5052
#include "utils/output_validation.hpp"
5153
#include "utils/sycl_alloc_utils.hpp"
5254
#include "utils/type_dispatch.hpp"
55+
#include "utils/type_utils.hpp"
5356

5457
namespace dpnp::extensions::indexing
5558
{
59+
namespace impl
60+
{
5661
namespace py = pybind11;
5762
namespace td_ns = dpctl::tensor::type_dispatch;
5863

@@ -504,16 +509,16 @@ void init_choose_dispatch_tables(void)
504509
init_dispatch_table<choose_fn_ptr_t, ChooseWrapFactory>(
505510
choose_wrap_dispatch_table);
506511
}
512+
} // namespace impl
507513

508514
void init_choose(py::module_ m)
509515
{
510-
dpnp::extensions::indexing::init_choose_dispatch_tables();
516+
impl::init_choose_dispatch_tables();
511517

512-
m.def("_choose", &py_choose, "", py::arg("src"), py::arg("chcs"),
518+
m.def("_choose", &impl::py_choose, "", py::arg("src"), py::arg("chcs"),
513519
py::arg("dst"), py::arg("mode"), py::arg("sycl_queue"),
514520
py::arg("depends") = py::list());
515521

516522
return;
517523
}
518-
519524
} // namespace dpnp::extensions::indexing

dpnp/backend/kernels/indexing/choose.hpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@
2828

2929
#pragma once
3030

31+
#include <cstddef>
32+
3133
#include <sycl/sycl.hpp>
3234

3335
#include "kernels/dpctl_tensor_types.hpp"

0 commit comments

Comments
 (0)