From b23fc4ce8200490015b05439fb34a49927021942 Mon Sep 17 00:00:00 2001
From: Malte Foerster <mfoerster@nvidia.com>
Date: Fri, 28 Oct 2022 13:51:30 -0700
Subject: [PATCH] hotfix linewiseop padded span test

---
 cpp/test/matrix/linewise_op.cu | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/cpp/test/matrix/linewise_op.cu b/cpp/test/matrix/linewise_op.cu
index 931c3ecb16..c61af89bec 100644
--- a/cpp/test/matrix/linewise_op.cu
+++ b/cpp/test/matrix/linewise_op.cu
@@ -108,7 +108,7 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
                             const T* vec)
   {
     auto f        = [] __device__(T a, T b) -> T { return a + b; };
-    auto vec_view = raft::make_device_vector_view<const T, I>(vec, lineLen);
+    auto vec_view = raft::make_device_vector_view<const T, I>(vec, alongLines ? lineLen : nLines);
     matrix::linewise_op(handle, in, out, alongLines, f, vec_view);
   }
 
@@ -234,8 +234,8 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
         common::nvtx::range dims_scope("Dims-%zu-%zu", std::size_t(n), std::size_t(m));
         common::nvtx::range dir_scope(alongRows ? "alongRows" : "acrossRows");
 
-        auto lineLen = m;
-        auto nLines  = n;
+        auto lineLen = alongRows ? m : n;
+        auto nLines  = alongRows ? n : m;
 
         // create a padded span based on testdata (just for functional testing)
         size_t matrix_size_padded;
@@ -244,7 +244,7 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
           typename raft::layout_right_padded<T>::mapping<matrix_extent<I>> layout{extents};
           matrix_size_padded = layout.required_span_size();
         } else {
-          auto extents = matrix_extent<I>{m, n};
+          auto extents = matrix_extent<I>{n, m};
           typename raft::layout_left_padded<T>::mapping<matrix_extent<I>> layout{extents};
           matrix_size_padded = layout.required_span_size();
         }
@@ -259,9 +259,9 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
           common::nvtx::range vecs_scope("one vec");
           if (alongRows) {
             auto inSpan = make_device_aligned_matrix_view<T, I, raft::layout_right_padded<T>>(
-              blob_in.data(), nLines, lineLen);
+              blob_in.data(), n, m);
             auto outSpan = make_device_aligned_matrix_view<T, I, raft::layout_right_padded<T>>(
-              blob_out.data(), nLines, lineLen);
+              blob_out.data(), n, m);
             // prep padded input data
             thrust::for_each_n(rmm::exec_policy(stream),
                                thrust::make_counting_iterator(0ul),
@@ -271,7 +271,7 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
                                });
             auto inSpanConst =
               make_device_aligned_matrix_view<const T, I, raft::layout_right_padded<T>>(
-                blob_in.data(), nLines, lineLen);
+                blob_in.data(), n, m);
             runLinewiseSumPadded<raft::layout_right_padded<T>>(
               outSpan, inSpanConst, lineLen, nLines, alongRows, vec1);
 
@@ -292,9 +292,9 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
 
           } else {
             auto inSpan = make_device_aligned_matrix_view<T, I, raft::layout_left_padded<T>>(
-              blob_in.data(), lineLen, nLines);
+              blob_in.data(), n, m);
             auto outSpan = make_device_aligned_matrix_view<T, I, raft::layout_left_padded<T>>(
-              blob_out.data(), lineLen, nLines);
+              blob_out.data(), n, m);
             // prep padded input data
             thrust::for_each_n(rmm::exec_policy(stream),
                                thrust::make_counting_iterator(0ul),
@@ -304,7 +304,7 @@ struct LinewiseTest : public ::testing::TestWithParam<typename ParamsReader::Par
                                });
             auto inSpanConst =
               make_device_aligned_matrix_view<const T, I, raft::layout_left_padded<T>>(
-                blob_in.data(), lineLen, nLines);
+                blob_in.data(), n, m);
             runLinewiseSumPadded<raft::layout_left_padded<T>>(
               outSpan, inSpanConst, lineLen, nLines, alongRows, vec1);