Skip to content

Commit e6e179e

Browse files
Clean-up includes
1 parent ca295f1 commit e6e179e

File tree

8 files changed

+17
-7
lines changed

8 files changed

+17
-7
lines changed

dpctl_ext/tensor/libtensor/source/elementwise_functions/add.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "add.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/add.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/atan2.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "atan2.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/atan2.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/bitwise_and.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "bitwise_and.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/bitwise_and.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/bitwise_left_shift.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "bitwise_left_shift.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/bitwise_left_shift.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/bitwise_or.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "bitwise_or.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/bitwise_or.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/bitwise_right_shift.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "bitwise_right_shift.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/bitwise_right_shift.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/bitwise_xor.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
#include "bitwise_xor.hpp"
4646
#include "elementwise_functions.hpp"
4747
#include "utils/type_dispatch.hpp"
48+
#include "utils/type_dispatch_building.hpp"
4849

4950
#include "kernels/elementwise_functions/bitwise_xor.hpp"
5051
#include "kernels/elementwise_functions/common.hpp"

dpctl_ext/tensor/libtensor/source/elementwise_functions/elementwise_functions.hpp

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@
4949
#include <pybind11/pybind11.h>
5050

5151
#include "elementwise_functions_type_utils.hpp"
52+
#include "kernels/alignment.hpp"
5253
#include "kernels/dpctl_tensor_types.hpp"
5354
#include "simplify_iteration_space.hpp"
5455
#include "utils/memory_overlap.hpp"
@@ -65,6 +66,9 @@ namespace dpctl::tensor::py_internal
6566
namespace py = pybind11;
6667
namespace td_ns = dpctl::tensor::type_dispatch;
6768

69+
using dpctl::tensor::kernels::alignment_utils::is_aligned;
70+
using dpctl::tensor::kernels::alignment_utils::required_alignment;
71+
6872
/*! @brief Template implementing Python API for unary elementwise functions */
6973
template <typename output_typesT,
7074
typename contig_dispatchT,
@@ -428,7 +432,7 @@ std::pair<sycl::event, sycl::event> py_binary_ufunc(
428432
int nd = dst_nd;
429433
const py::ssize_t *shape = src1_shape;
430434

431-
dpctl::tensor::py_internal::simplify_iteration_space_3(
435+
simplify_iteration_space_3(
432436
nd, shape, src1_strides, src2_strides, dst_strides,
433437
// outputs
434438
simplified_shape, simplified_src1_strides, simplified_src2_strides,
@@ -597,7 +601,7 @@ py::object py_binary_ufunc_result_type(const py::dtype &input1_dtype,
597601
return py::cast<py::object>(res);
598602
}
599603
else {
600-
using dpctl::tensor::py_internal::type_utils::_dtype_from_typenum;
604+
using type_utils::_dtype_from_typenum;
601605

602606
auto dst_typenum_t = static_cast<td_ns::typenum_t>(dst_typeid);
603607
auto dt = _dtype_from_typenum(dst_typenum_t);
@@ -724,11 +728,10 @@ std::pair<sycl::event, sycl::event>
724728
int nd = lhs_nd;
725729
const py::ssize_t *shape = rhs_shape;
726730

727-
dpctl::tensor::py_internal::simplify_iteration_space(
728-
nd, shape, rhs_strides, lhs_strides,
729-
// outputs
730-
simplified_shape, simplified_rhs_strides, simplified_lhs_strides,
731-
rhs_offset, lhs_offset);
731+
simplify_iteration_space(nd, shape, rhs_strides, lhs_strides,
732+
// outputs
733+
simplified_shape, simplified_rhs_strides,
734+
simplified_lhs_strides, rhs_offset, lhs_offset);
732735

733736
std::vector<sycl::event> host_tasks{};
734737
if (nd < 3) {

0 commit comments

Comments
 (0)