|
30 | 30 | #include <cstddef> |
31 | 31 | #include <cstdint> |
32 | 32 | #include <memory> |
| 33 | +#include <stdexcept> |
| 34 | +#include <string> |
33 | 35 | #include <type_traits> |
34 | 36 | #include <utility> |
35 | 37 | #include <vector> |
|
46 | 48 | // dpctl tensor headers |
47 | 49 | #include "utils/indexing_utils.hpp" |
48 | 50 | #include "utils/memory_overlap.hpp" |
49 | | -#include "utils/offset_utils.hpp" // |
| 51 | +#include "utils/offset_utils.hpp" |
50 | 52 | #include "utils/output_validation.hpp" |
51 | 53 | #include "utils/sycl_alloc_utils.hpp" |
52 | 54 | #include "utils/type_dispatch.hpp" |
| 55 | +#include "utils/type_utils.hpp" |
53 | 56 |
|
54 | 57 | namespace dpnp::extensions::indexing |
55 | 58 | { |
| 59 | +namespace impl |
| 60 | +{ |
56 | 61 | namespace py = pybind11; |
57 | 62 | namespace td_ns = dpctl::tensor::type_dispatch; |
58 | 63 |
|
@@ -504,16 +509,16 @@ void init_choose_dispatch_tables(void) |
504 | 509 | init_dispatch_table<choose_fn_ptr_t, ChooseWrapFactory>( |
505 | 510 | choose_wrap_dispatch_table); |
506 | 511 | } |
| 512 | +} // namespace impl |
507 | 513 |
|
508 | 514 | void init_choose(py::module_ m) |
509 | 515 | { |
510 | | - dpnp::extensions::indexing::init_choose_dispatch_tables(); |
| 516 | + impl::init_choose_dispatch_tables(); |
511 | 517 |
|
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"), |
513 | 519 | py::arg("dst"), py::arg("mode"), py::arg("sycl_queue"), |
514 | 520 | py::arg("depends") = py::list()); |
515 | 521 |
|
516 | 522 | return; |
517 | 523 | } |
518 | | - |
519 | 524 | } // namespace dpnp::extensions::indexing |
0 commit comments