Skip to content

Commit b9406ed

Browse files
committed
Update indexing extension to use init_dispatch_table() from common utils
1 parent 7a92951 commit b9406ed

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

dpnp/backend/extensions/indexing/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,8 @@ endif()
5757

5858
set_target_properties(${python_module_name} PROPERTIES CMAKE_POSITION_INDEPENDENT_CODE ON)
5959

60-
target_include_directories(${python_module_name} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../../include)
61-
target_include_directories(${python_module_name} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../../src)
60+
target_include_directories(${python_module_name} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../../)
61+
target_include_directories(${python_module_name} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../common)
6262

6363
target_include_directories(${python_module_name} PUBLIC ${Dpctl_INCLUDE_DIR})
6464
target_include_directories(${python_module_name} PUBLIC ${Dpctl_TENSOR_INCLUDE_DIR})

dpnp/backend/extensions/indexing/choose.cpp

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,11 @@
3636

3737
#include "choose_kernel.hpp"
3838
#include "dpctl4pybind11.hpp"
39+
40+
// utils extension header
41+
#include "ext/common.hpp"
42+
43+
// dpctl tensor headers
3944
#include "utils/indexing_utils.hpp"
4045
#include "utils/memory_overlap.hpp"
4146
#include "utils/output_validation.hpp"
@@ -432,18 +437,13 @@ using ChooseClipFactory = ChooseFactory<fnT, IndT, T, ClipIndex<IndT>>;
432437

433438
void init_choose_dispatch_tables(void)
434439
{
435-
using namespace td_ns;
440+
using ext::common::init_dispatch_table;
436441
using kernels::choose_fn_ptr_t;
437442

438-
DispatchTableBuilder<choose_fn_ptr_t, ChooseClipFactory, num_types>
439-
dtb_choose_clip;
440-
dtb_choose_clip.populate_dispatch_table(choose_clip_dispatch_table);
441-
442-
DispatchTableBuilder<choose_fn_ptr_t, ChooseWrapFactory, num_types>
443-
dtb_choose_wrap;
444-
dtb_choose_wrap.populate_dispatch_table(choose_wrap_dispatch_table);
445-
446-
return;
443+
init_dispatch_table<choose_fn_ptr_t, ChooseClipFactory>(
444+
choose_clip_dispatch_table);
445+
init_dispatch_table<choose_fn_ptr_t, ChooseWrapFactory>(
446+
choose_wrap_dispatch_table);
447447
}
448448

449449
void init_choose(py::module_ m)

0 commit comments

Comments
 (0)