Added test_top_k_per_row to test-pipeline.yaml. (#26569)

Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>

.buildkite/test-pipeline.yaml

@@ -431,8 +431,9 @@ steps:
   source_file_dependencies:
   - csrc/
   - tests/kernels/core
+  - tests/kernels/test_top_k_per_row.py
   commands:
-    - pytest -v -s kernels/core
+    - pytest -v -s kernels/core kernels/test_top_k_per_row.py
 
 - label: Kernels Attention Test %N # 23min
   timeout_in_minutes: 35

tests/kernels/test_top_k_per_row.py

@@ -125,7 +125,6 @@ def test_top_k_per_row(
         indices,
         values,
         num_rows,
-        top_k,
         logits.stride(0),
         logits.stride(1),
     )