From 165cec3c0bf8b8ab6bf981488cfe823bc6e69baa Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Tue, 18 Apr 2023 10:58:13 -0400 Subject: [PATCH 1/2] Removing cuda stream view include from mdarray --- cpp/include/raft/core/mdarray.hpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cpp/include/raft/core/mdarray.hpp b/cpp/include/raft/core/mdarray.hpp index 61c1b500e6..ae8ad5eb0f 100644 --- a/cpp/include/raft/core/mdarray.hpp +++ b/cpp/include/raft/core/mdarray.hpp @@ -29,7 +29,6 @@ #include #include #include -#include namespace raft { /** @@ -45,11 +44,11 @@ namespace raft { template class array_interface { /** - * @brief Get a mdspan that can be passed down to CUDA kernels. + * @brief Get an mdspan */ auto view() noexcept { return static_cast(this)->view(); } /** - * @brief Get a mdspan that can be passed down to CUDA kernels. + * @brief Get a const mdspan */ auto view() const noexcept { return static_cast(this)->view(); } }; @@ -108,7 +107,8 @@ inline constexpr bool is_array_interface_v = is_array_interface::value; * template. * * - Most of the constructors from the reference implementation is removed to make sure - * CUDA stream is honorred. + * CUDA stream is honored. Note that this class is not coupled to CUDA and therefore + * will only be used in the case where the device variant is used. * * - unique_size is not implemented, which is still working in progress in the proposal * @@ -220,11 +220,11 @@ class mdarray #undef RAFT_MDARRAY_CTOR_CONSTEXPR /** - * @brief Get a mdspan that can be passed down to CUDA kernels. + * @brief Get an mdspan */ auto view() noexcept { return view_type(c_.data(), map_, cp_.make_accessor_policy()); } /** - * @brief Get a mdspan that can be passed down to CUDA kernels. + * @brief Get a const mdspan */ auto view() const noexcept { From ef9000981815a1ac64233f4a1155f0e9b62db739 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Tue, 18 Apr 2023 12:53:21 -0400 Subject: [PATCH 2/2] Fixing comments --- cpp/include/raft/core/mdarray.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cpp/include/raft/core/mdarray.hpp b/cpp/include/raft/core/mdarray.hpp index ae8ad5eb0f..88f90485dd 100644 --- a/cpp/include/raft/core/mdarray.hpp +++ b/cpp/include/raft/core/mdarray.hpp @@ -48,7 +48,7 @@ class array_interface { */ auto view() noexcept { return static_cast(this)->view(); } /** - * @brief Get a const mdspan + * @brief Get an mdspan */ auto view() const noexcept { return static_cast(this)->view(); } }; @@ -224,7 +224,7 @@ class mdarray */ auto view() noexcept { return view_type(c_.data(), map_, cp_.make_accessor_policy()); } /** - * @brief Get a const mdspan + * @brief Get an mdspan */ auto view() const noexcept {