Avoid illegal access for degenerated subslices of length 0

kokkos · Jun 10, 2024 · 8244ef2 · 8244ef2
1 parent 5eb2aa5
commit 8244ef2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_submdspan.cpp b/tests/test_submdspan.cpp
@@ -262,6 +262,7 @@ struct TestSubMDSpan<
     } else if constexpr (SubMDSpan::rank() == 0) {
       return (&src_mds[SrcIdx...]==&sub_mds[]); 
     } else {
+      if(sub_mds.size() == 0) return true;
       return (&src_mds[SrcIdx...]==&sub_mds[SubIdx...]);
     }
 #else
@@ -270,6 +271,7 @@ struct TestSubMDSpan<
     } else if constexpr (SubMDSpan::rank() == 0) {
       return (&src_mds(SrcIdx...)==&sub_mds());
     } else {
+      if(sub_mds.size() == 0) return true;
       return (&src_mds(SrcIdx...)==&sub_mds(SubIdx...));
     }
 #endif