Fixing an index error introduced in PR #1109 (#1110)

My PR #1109 has an index error, pointed out by @benfred in [this](#1109 (comment)) comment. Authors: - Vinay Deshpande (https://github.com/vinaydes) Approvers: - Ben Frederickson (https://github.com/benfred) - Corey J. Nolet (https://github.com/cjnolet) URL: #1110
rapidsai · Dec 21, 2022 · 0612284 · 0612284
1 parent aec0948
commit 0612284
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/test/random/sample_without_replacement.cu b/cpp/test/random/sample_without_replacement.cu
@@ -217,7 +217,7 @@ const std::vector<SWoRInputs<float>> inputsf = {{1024, 512, -1, 0.f, GenPhilox,
     if (params.largeWeightIndex >= 0) {                                                            \
       ASSERT_TRUE((h_outIdx[0] == params.largeWeightIndex) ||                                      \
                   (h_outIdx[1] == params.largeWeightIndex) ||                                      \
-                  (h_outIdx[3] == params.largeWeightIndex));                                       \
+                  (h_outIdx[2] == params.largeWeightIndex));                                       \
     }                                                                                              \
   } while (false)