3333// utils extension header
3434#include " ext/common.hpp"
3535
36- // dpctl tensor headers
36+ // dpnp tensor headers
3737#include " utils/memory_overlap.hpp"
3838#include " utils/output_validation.hpp"
3939#include " utils/type_utils.hpp"
@@ -45,7 +45,7 @@ namespace dpnp::extensions::blas
4545{
4646namespace mkl_blas = oneapi::mkl::blas;
4747namespace py = pybind11;
48- namespace type_utils = dpctl ::tensor::type_utils;
48+ namespace type_utils = dpnp ::tensor::type_utils;
4949
5050using ext::common::init_dispatch_table;
5151
@@ -70,7 +70,7 @@ typedef sycl::event (*gemm_batch_impl_fn_ptr_t)(
7070 const std::vector<sycl::event> &);
7171
7272static gemm_batch_impl_fn_ptr_t
73- gemm_batch_dispatch_table[dpctl_td_ns ::num_types][dpctl_td_ns ::num_types];
73+ gemm_batch_dispatch_table[dpnp_td_ns ::num_types][dpnp_td_ns ::num_types];
7474
7575template <typename Tab, typename Tc>
7676static sycl::event gemm_batch_impl (sycl::queue &exec_q,
@@ -212,9 +212,9 @@ void standardize_strides_to_zero(std::vector<py::ssize_t> &strides,
212212
213213std::tuple<sycl::event, sycl::event, bool >
214214 gemm_batch (sycl::queue &exec_q,
215- const dpctl ::tensor::usm_ndarray &matrixA,
216- const dpctl ::tensor::usm_ndarray &matrixB,
217- const dpctl ::tensor::usm_ndarray &resultC,
215+ const dpnp ::tensor::usm_ndarray &matrixA,
216+ const dpnp ::tensor::usm_ndarray &matrixB,
217+ const dpnp ::tensor::usm_ndarray &resultC,
218218 const std::vector<sycl::event> &depends = {})
219219{
220220 const int matrixA_nd = matrixA.get_ndim ();
@@ -225,7 +225,7 @@ std::tuple<sycl::event, sycl::event, bool>
225225 throw py::value_error (" The given arrays have incorrect dimensions." );
226226 }
227227
228- auto const &overlap = dpctl ::tensor::overlap::MemoryOverlap ();
228+ auto const &overlap = dpnp ::tensor::overlap::MemoryOverlap ();
229229 if (overlap (matrixA, resultC)) {
230230 throw py::value_error (" Input array 1 and output array are overlapping "
231231 " segments of memory" );
@@ -235,7 +235,7 @@ std::tuple<sycl::event, sycl::event, bool>
235235 " segments of memory" );
236236 }
237237
238- if (!dpctl ::utils::queues_are_compatible (
238+ if (!dpnp ::utils::queues_are_compatible (
239239 exec_q,
240240 {matrixA.get_queue (), matrixB.get_queue (), resultC.get_queue ()})) {
241241 throw py::value_error (
@@ -262,9 +262,9 @@ std::tuple<sycl::event, sycl::event, bool>
262262 " the number of columns in result array." );
263263 }
264264 const std::int64_t src_nelems = batch_size * m * n;
265- dpctl ::tensor::validation::CheckWritable::throw_if_not_writable (resultC);
266- dpctl ::tensor::validation::AmpleMemory::throw_if_not_ample (resultC,
267- src_nelems);
265+ dpnp ::tensor::validation::CheckWritable::throw_if_not_writable (resultC);
266+ dpnp ::tensor::validation::AmpleMemory::throw_if_not_ample (resultC,
267+ src_nelems);
268268
269269 std::vector<py::ssize_t > a_stride = matrixA.get_strides_vector ();
270270 std::vector<py::ssize_t > b_stride = matrixB.get_strides_vector ();
@@ -374,7 +374,7 @@ std::tuple<sycl::event, sycl::event, bool>
374374 throw py::value_error (" matrixA and matrixB must be of the same type." );
375375 }
376376
377- auto array_types = dpctl_td_ns ::usm_ndarray_types ();
377+ auto array_types = dpnp_td_ns ::usm_ndarray_types ();
378378 const int matrixAB_type_id =
379379 array_types.typenum_to_lookup_id (matrixA_typenum);
380380 const int resultC_type_id =
@@ -397,7 +397,7 @@ std::tuple<sycl::event, sycl::event, bool>
397397 strideb, stridec, transA, transB, a_typeless_ptr,
398398 b_typeless_ptr, r_typeless_ptr, is_row_major, depends);
399399
400- sycl::event args_ev = dpctl ::utils::keep_args_alive (
400+ sycl::event args_ev = dpnp ::utils::keep_args_alive (
401401 exec_q, {matrixA, matrixB, resultC}, {gemm_batch_ev});
402402
403403 return std::make_tuple (args_ev, gemm_batch_ev, is_row_major);
0 commit comments