Fix loop variable in token search for top_k indices

Author: hariharans29

test/sampling_tests.cpp

@@ -697,8 +697,8 @@ TEST(SamplingTests, RandomizedSamplingSelectTopCuda_BatchSize1_LargeVocabSize) {
 
       // Next token has to be in the list of the above top_k indices
       bool found = false;
-      for (int i = 0; i < top_k; ++i) {
-        if (indices[i] == next_token) {
+      for (int k = 0; k < top_k; ++k) {
+        if (indices[k] == next_token) {
           found = true;
           break;
         }