From 0612284f0a30d689a5ed38039fb2385712a9bb5f Mon Sep 17 00:00:00 2001 From: Vinay Deshpande Date: Wed, 21 Dec 2022 06:00:36 +0530 Subject: [PATCH] Fixing an index error introduced in PR #1109 (#1110) My PR #1109 has an index error, pointed out by @benfred in [this](https://github.com/rapidsai/raft/pull/1109#discussion_r1051329480) comment. Authors: - Vinay Deshpande (https://github.com/vinaydes) Approvers: - Ben Frederickson (https://github.com/benfred) - Corey J. Nolet (https://github.com/cjnolet) URL: https://github.com/rapidsai/raft/pull/1110 --- cpp/test/random/sample_without_replacement.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/test/random/sample_without_replacement.cu b/cpp/test/random/sample_without_replacement.cu index e222f28ce6..c9b4dd0879 100644 --- a/cpp/test/random/sample_without_replacement.cu +++ b/cpp/test/random/sample_without_replacement.cu @@ -217,7 +217,7 @@ const std::vector> 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)