Build fix

microsoft · Nov 14, 2024 · b90bfc6 · b90bfc6
1 parent 4e7c63d
commit b90bfc6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/sampling_benchmark.cpp b/test/sampling_benchmark.cpp
@@ -144,8 +144,9 @@ TEST_P(SamplingBenchmarkTest, RunBenchmark) {
 auto benchmark_values = ::testing::Values(
     BenchmarkParams{Generators::DeviceType::CPU, 1, BenchmarkFunction::TopP},
     BenchmarkParams{Generators::DeviceType::CPU, 1, BenchmarkFunction::TopK},
-    BenchmarkParams{Generators::DeviceType::CPU, 1, BenchmarkFunction::TopKTopP},
+    BenchmarkParams{Generators::DeviceType::CPU, 1, BenchmarkFunction::TopKTopP}
 #if USE_CUDA
+    ,
     BenchmarkParams{Generators::DeviceType::CUDA, 1, BenchmarkFunction::TopP},
     BenchmarkParams{Generators::DeviceType::CUDA, 1, BenchmarkFunction::TopK},
     BenchmarkParams{Generators::DeviceType::CUDA, 1, BenchmarkFunction::TopKTopP},