@@ -93,27 +93,27 @@ void _split_iteration_space(const shT &shape_vec,
93
93
94
94
// Computation of positions of masked elements
95
95
96
+ namespace td_ns = dpctl::tensor::type_dispatch;
97
+
96
98
using dpctl::tensor::kernels::indexing::mask_positions_contig_impl_fn_ptr_t ;
97
99
static mask_positions_contig_impl_fn_ptr_t
98
- mask_positions_contig_dispatch_vector[dpctl::tensor::detail ::num_types];
100
+ mask_positions_contig_dispatch_vector[td_ns ::num_types];
99
101
100
102
using dpctl::tensor::kernels::indexing::mask_positions_strided_impl_fn_ptr_t ;
101
103
static mask_positions_strided_impl_fn_ptr_t
102
- mask_positions_strided_dispatch_vector[dpctl::tensor::detail ::num_types];
104
+ mask_positions_strided_dispatch_vector[td_ns ::num_types];
103
105
104
106
void populate_mask_positions_dispatch_vectors (void )
105
107
{
106
108
using dpctl::tensor::kernels::indexing::MaskPositionsContigFactory;
107
- dpctl::tensor::detail::DispatchVectorBuilder<
108
- mask_positions_contig_impl_fn_ptr_t , MaskPositionsContigFactory,
109
- dpctl::tensor::detail::num_types>
109
+ td_ns::DispatchVectorBuilder<mask_positions_contig_impl_fn_ptr_t ,
110
+ MaskPositionsContigFactory, td_ns::num_types>
110
111
dvb1;
111
112
dvb1.populate_dispatch_vector (mask_positions_contig_dispatch_vector);
112
113
113
114
using dpctl::tensor::kernels::indexing::MaskPositionsStridedFactory;
114
- dpctl::tensor::detail::DispatchVectorBuilder<
115
- mask_positions_strided_impl_fn_ptr_t , MaskPositionsStridedFactory,
116
- dpctl::tensor::detail::num_types>
115
+ td_ns::DispatchVectorBuilder<mask_positions_strided_impl_fn_ptr_t ,
116
+ MaskPositionsStridedFactory, td_ns::num_types>
117
117
dvb2;
118
118
dvb2.populate_dispatch_vector (mask_positions_strided_dispatch_vector);
119
119
@@ -158,14 +158,13 @@ size_t py_mask_positions(dpctl::tensor::usm_ndarray mask,
158
158
const char *mask_data = mask.get_data ();
159
159
char *cumsum_data = cumsum.get_data ();
160
160
161
- auto const &array_types = dpctl::tensor::detail ::usm_ndarray_types ();
161
+ auto const &array_types = td_ns ::usm_ndarray_types ();
162
162
163
163
int mask_typeid = array_types.typenum_to_lookup_id (mask_typenum);
164
164
int cumsum_typeid = array_types.typenum_to_lookup_id (cumsum_typenum);
165
165
166
166
// cumsum must be int64_t only
167
- constexpr int int64_typeid =
168
- static_cast <int >(dpctl::tensor::detail::typenum_t ::INT64);
167
+ constexpr int int64_typeid = static_cast <int >(td_ns::typenum_t ::INT64);
169
168
if (cumsum_typeid != int64_typeid) {
170
169
throw py::value_error (
171
170
" Cumulative sum array must have int64 data-type." );
@@ -244,30 +243,28 @@ using dpctl::tensor::kernels::indexing::
244
243
masked_extract_all_slices_strided_impl_fn_ptr_t ;
245
244
246
245
static masked_extract_all_slices_strided_impl_fn_ptr_t
247
- masked_extract_all_slices_strided_impl_dispatch_vector
248
- [dpctl::tensor::detail::num_types];
246
+ masked_extract_all_slices_strided_impl_dispatch_vector[td_ns::num_types];
249
247
250
248
using dpctl::tensor::kernels::indexing::
251
249
masked_extract_some_slices_strided_impl_fn_ptr_t ;
252
250
253
251
static masked_extract_some_slices_strided_impl_fn_ptr_t
254
- masked_extract_some_slices_strided_impl_dispatch_vector
255
- [dpctl::tensor::detail::num_types];
252
+ masked_extract_some_slices_strided_impl_dispatch_vector[td_ns::num_types];
256
253
257
254
void populate_masked_extract_dispatch_vectors (void )
258
255
{
259
256
using dpctl::tensor::kernels::indexing::MaskExtractAllSlicesStridedFactory;
260
- dpctl::tensor::detail ::DispatchVectorBuilder<
257
+ td_ns ::DispatchVectorBuilder<
261
258
masked_extract_all_slices_strided_impl_fn_ptr_t ,
262
- MaskExtractAllSlicesStridedFactory, dpctl::tensor::detail ::num_types>
259
+ MaskExtractAllSlicesStridedFactory, td_ns ::num_types>
263
260
dvb1;
264
261
dvb1.populate_dispatch_vector (
265
262
masked_extract_all_slices_strided_impl_dispatch_vector);
266
263
267
264
using dpctl::tensor::kernels::indexing::MaskExtractSomeSlicesStridedFactory;
268
- dpctl::tensor::detail ::DispatchVectorBuilder<
265
+ td_ns ::DispatchVectorBuilder<
269
266
masked_extract_some_slices_strided_impl_fn_ptr_t ,
270
- MaskExtractSomeSlicesStridedFactory, dpctl::tensor::detail ::num_types>
267
+ MaskExtractSomeSlicesStridedFactory, td_ns ::num_types>
271
268
dvb2;
272
269
dvb2.populate_dispatch_vector (
273
270
masked_extract_some_slices_strided_impl_dispatch_vector);
@@ -359,13 +356,12 @@ py_extract(dpctl::tensor::usm_ndarray src,
359
356
int dst_typenum = dst.get_typenum ();
360
357
int cumsum_typenum = cumsum.get_typenum ();
361
358
362
- auto const &array_types = dpctl::tensor::detail ::usm_ndarray_types ();
359
+ auto const &array_types = td_ns ::usm_ndarray_types ();
363
360
int src_typeid = array_types.typenum_to_lookup_id (src_typenum);
364
361
int dst_typeid = array_types.typenum_to_lookup_id (dst_typenum);
365
362
int cumsum_typeid = array_types.typenum_to_lookup_id (cumsum_typenum);
366
363
367
- constexpr int int64_typeid =
368
- static_cast <int >(dpctl::tensor::detail::typenum_t ::INT64);
364
+ constexpr int int64_typeid = static_cast <int >(td_ns::typenum_t ::INT64);
369
365
if (cumsum_typeid != int64_typeid) {
370
366
throw py::value_error (
371
367
" Unexact data type of cumsum array, expecting 'int64'" );
@@ -557,30 +553,28 @@ using dpctl::tensor::kernels::indexing::
557
553
masked_place_all_slices_strided_impl_fn_ptr_t ;
558
554
559
555
static masked_place_all_slices_strided_impl_fn_ptr_t
560
- masked_place_all_slices_strided_impl_dispatch_vector
561
- [dpctl::tensor::detail::num_types];
556
+ masked_place_all_slices_strided_impl_dispatch_vector[td_ns::num_types];
562
557
563
558
using dpctl::tensor::kernels::indexing::
564
559
masked_place_some_slices_strided_impl_fn_ptr_t ;
565
560
566
561
static masked_place_some_slices_strided_impl_fn_ptr_t
567
- masked_place_some_slices_strided_impl_dispatch_vector
568
- [dpctl::tensor::detail::num_types];
562
+ masked_place_some_slices_strided_impl_dispatch_vector[td_ns::num_types];
569
563
570
564
void populate_masked_place_dispatch_vectors (void )
571
565
{
572
566
using dpctl::tensor::kernels::indexing::MaskPlaceAllSlicesStridedFactory;
573
- dpctl::tensor::detail:: DispatchVectorBuilder<
574
- masked_place_all_slices_strided_impl_fn_ptr_t ,
575
- MaskPlaceAllSlicesStridedFactory, dpctl::tensor::detail ::num_types>
567
+ td_ns:: DispatchVectorBuilder<masked_place_all_slices_strided_impl_fn_ptr_t ,
568
+ MaskPlaceAllSlicesStridedFactory ,
569
+ td_ns ::num_types>
576
570
dvb1;
577
571
dvb1.populate_dispatch_vector (
578
572
masked_place_all_slices_strided_impl_dispatch_vector);
579
573
580
574
using dpctl::tensor::kernels::indexing::MaskPlaceSomeSlicesStridedFactory;
581
- dpctl::tensor::detail:: DispatchVectorBuilder<
582
- masked_place_some_slices_strided_impl_fn_ptr_t ,
583
- MaskPlaceSomeSlicesStridedFactory, dpctl::tensor::detail ::num_types>
575
+ td_ns:: DispatchVectorBuilder<masked_place_some_slices_strided_impl_fn_ptr_t ,
576
+ MaskPlaceSomeSlicesStridedFactory ,
577
+ td_ns ::num_types>
584
578
dvb2;
585
579
dvb2.populate_dispatch_vector (
586
580
masked_place_some_slices_strided_impl_dispatch_vector);
@@ -673,13 +667,12 @@ py_place(dpctl::tensor::usm_ndarray dst,
673
667
int rhs_typenum = rhs.get_typenum ();
674
668
int cumsum_typenum = cumsum.get_typenum ();
675
669
676
- auto const &array_types = dpctl::tensor::detail ::usm_ndarray_types ();
670
+ auto const &array_types = td_ns ::usm_ndarray_types ();
677
671
int dst_typeid = array_types.typenum_to_lookup_id (dst_typenum);
678
672
int rhs_typeid = array_types.typenum_to_lookup_id (rhs_typenum);
679
673
int cumsum_typeid = array_types.typenum_to_lookup_id (cumsum_typenum);
680
674
681
- constexpr int int64_typeid =
682
- static_cast <int >(dpctl::tensor::detail::typenum_t ::INT64);
675
+ constexpr int int64_typeid = static_cast <int >(td_ns::typenum_t ::INT64);
683
676
if (cumsum_typeid != int64_typeid) {
684
677
throw py::value_error (
685
678
" Unexact data type of cumsum array, expecting 'int64'" );
@@ -913,15 +906,14 @@ std::pair<sycl::event, sycl::event> py_nonzero(
913
906
py::ssize_t nz_elems = indexes_shape[1 ];
914
907
915
908
int indexes_typenum = indexes.get_typenum ();
916
- auto const &array_types = dpctl::tensor::detail ::usm_ndarray_types ();
909
+ auto const &array_types = td_ns ::usm_ndarray_types ();
917
910
int indexes_typeid = array_types.typenum_to_lookup_id (indexes_typenum);
918
911
919
912
int cumsum_typenum = cumsum.get_typenum ();
920
913
int cumsum_typeid = array_types.typenum_to_lookup_id (cumsum_typenum);
921
914
922
915
// cumsum must be int64_t only
923
- constexpr int int64_typeid =
924
- static_cast <int >(dpctl::tensor::detail::typenum_t ::INT64);
916
+ constexpr int int64_typeid = static_cast <int >(td_ns::typenum_t ::INT64);
925
917
if (cumsum_typeid != int64_typeid || indexes_typeid != int64_typeid) {
926
918
throw py::value_error (
927
919
" Cumulative sum array and index array must have int64 data-type" );
0 commit comments