From fc229379568398f91759fa957539340a0b48c212 Mon Sep 17 00:00:00 2001 From: Vinay D Date: Sat, 17 Dec 2022 17:54:57 +0530 Subject: [PATCH] Fixing the index of sorted array --- 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)