fix repeat_interleave benchmark bug

benchmark/conftest.py

@@ -177,11 +177,12 @@ def setup_once(request):
         print("\nThis is query mode; all benchmark functions will be skipped.")
     else:
         note_info = (
-        f"\n\nNote: The 'size' field below is for backward compatibility with previous versions of the benchmark. "
-        f"\nThis field will be removed in a future release."
+            f"\n\nNote: The 'size' field below is for backward compatibility with previous versions of the benchmark. "
+            f"\nThis field will be removed in a future release."
         )
         print(note_info)
 
+
 @pytest.fixture()
 def extract_and_log_op_attributes(request):
     print("")

benchmark/test_tensor_concat_perf.py

@@ -124,8 +124,20 @@ def repeat_input_fn(shape, cur_dtype, device):
 
 def repeat_interleave_self_input_fn(shape, dtype, device):
     inp = generate_tensor_input(shape, dtype, device)
-    repeats = 3
-    yield inp, repeats
+    repeats = torch.randint(
+        low=0,
+        high=0x2F,
+        size=[
+            shape[0],
+        ],
+        device=device,
+    )
+    dim = 0
+    # repeat_interleave.self_Tensor(Tensor self, Tensor repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor
+    yield inp, repeats, dim
+    if Config.bench_level == BenchLevel.COMPREHENSIVE:
+        # repeat_interleave.self_int(Tensor self, SymInt repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor
+        yield inp, 3
 
 
 @pytest.mark.parametrize(