kokkos#6805: fix host function call from device and fix more cuda war…

…nings in mdspan
nmm0 · Mar 7, 2024 · 56ffeb6 · 56ffeb6
1 parent caf9242
commit 56ffeb6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/core/src/View/MDSpan/Kokkos_MDSpan_Extents.hpp b/core/src/View/MDSpan/Kokkos_MDSpan_Extents.hpp
@@ -23,6 +23,7 @@ static_assert(false,
 #define KOKKOS_EXPERIMENTAL_MDSPAN_EXTENTS_HPP
 
 #include "Kokkos_MDSpan_Header.hpp"
+#include <Kokkos_Core_fwd.hpp>
 
 namespace Kokkos::Impl {
 
@@ -110,7 +111,7 @@ struct DataTypeFromExtents {
 /// Convert from a mdspan extent to a Kokkos extent, inserting 0s for static
 /// extents
 template <class Extents>
-auto dimension_from_extent(const Extents &e, std::size_t r) noexcept {
+KOKKOS_INLINE_FUNCTION auto dimension_from_extent(const Extents &e, std::size_t r) noexcept {
   return Extents::static_extent(r) == dynamic_extent ? e.extent(r) : 0;
 }
 }  // namespace Kokkos::Experimental::Impl

diff --git a/core/src/View/MDSpan/Kokkos_MDSpan_Layout.hpp b/core/src/View/MDSpan/Kokkos_MDSpan_Layout.hpp
@@ -59,7 +59,7 @@ struct ViewOffsetFromExtents {
 
 template <class ElementType, class Extents, class LayoutPolicy,
           class AccessorPolicy>
-auto array_layout_from_mdspan(
+KOKKOS_INLINE_FUNCTION auto array_layout_from_mdspan(
     const mdspan<ElementType, Extents, LayoutPolicy, AccessorPolicy> &mds) {
   using layout_type = typename ArrayLayoutFromLayout<LayoutPolicy>::type;
   const auto &ext   = mds.extents();
@@ -81,7 +81,7 @@ auto array_layout_from_mdspan(
 
 template <class ElementType, class Extents, class LayoutPolicy,
           class AccessorPolicy>
-auto view_offset_from_mdspan(
+KOKKOS_INLINE_FUNCTION auto view_offset_from_mdspan(
     const mdspan<ElementType, Extents, LayoutPolicy, AccessorPolicy> &mds) {
   using offset_type =
       typename ViewOffsetFromExtents<ElementType, Extents, LayoutPolicy>::type;

diff --git a/tpls/mdspan/include/experimental/__p2642_bits/layout_padded.hpp b/tpls/mdspan/include/experimental/__p2642_bits/layout_padded.hpp
@@ -110,6 +110,7 @@ struct padded_extent {
       return {find_next_multiple(pv,
                                    exts.extent(_ExtentToPadIdx))};
     } else {
+      (void)pv; // Some compilers complain about pv not being used here
       return {};
     }
   }