Update common/unit_test/Test_prep_transpose_view.cpp

Co-authored-by: yasahi-hpc <[email protected]>
kokkos · May 24, 2024 · 19bd366 · 19bd366
1 parent 8001d31
commit 19bd366
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/common/unit_test/Test_prep_transpose_view.cpp b/common/unit_test/Test_prep_transpose_view.cpp
@@ -142,7 +142,9 @@ void test_unmanaged_prep_transpose_view() {
     }
 
     LayoutType layout_orig;
-    layout_orig.dimension[0] = 1;
+    for (int i = 0; i < DIMS; ++i) {
+      layout_orig.dimension[i] = 1;
+    }
     InManagedViewType in("in", layout);
     OutManagedViewType out("out", layout_orig);
     OutViewType u_out(out.data(), layout_orig);