align the lint check

Author: RUIJIEZHONG66166

test/microbench/adaptive_avg_pool2d.py

@@ -1,5 +1,6 @@
 import time
 import argparse
+
 import torch
 from torch.profiler import profile, ProfilerActivity
 
@@ -48,21 +49,22 @@ def Adaptive_AVGPool2d(shape, dtype, channels_last, backward, device):
 def run_profile(shape, dtype, channels_last, backward, device, num_iter):
     with profile(
         activities=[ProfilerActivity.CPU,
-                  ProfilerActivity.XPU if device == 'xpu' else ProfilerActivity.CUDA],
+                  ProfilerActivity.XPU if device == "xpu" else ProfilerActivity.CUDA,
+        ],
         record_shapes=True,
     ) as prof:
         for i in range(num_iter):
             Adaptive_AVGPool2d(shape, dtype, channels_last, backward, device)
-    print(prof.key_averages().table(sort_by="{}_time_total".format(device)))
+    print(prof.key_averages().table(sort_by=f"{device}_time_total"))
 
 def run_e2e(shape, dtype, channels_last, backward, device, num_iter):
-    if device in ['xpu', 'cuda']:
-        torch.xpu.synchronize() if device == 'xpu' else torch.cuda.synchronize()
+    if device in ["xpu", "cuda"]:
+        torch.xpu.synchronize() if device == "xpu" else torch.cuda.synchronize()
     t1 = time.time()
     for i in range(num_iter):
         Adaptive_AVGPool2d(shape, dtype, channels_last, backward, device)
-    if device in ['xpu', 'cuda']:
-        torch.xpu.synchronize() if device == 'xpu' else torch.cuda.synchronize()
+    if device in ["xpu", "cuda"]:
+        torch.xpu.synchronize() if device == "xpu" else torch.cuda.synchronize()
     t2 = time.time()
     e2e_time = (t2 - t1) / num_iter
     print("E2E total time:", f"{float(e2e_time):.20f}")
@@ -88,22 +90,39 @@ def benchmark(args):
                     backward,
                 )
                 if not args.e2e_only:
-                    run_profile(shape, dtype, channels_last, backward, args.device, args.num_iter)
+                    run_profile(
+                        shape,
+                        dtype,
+                        channels_last,
+                        backward,
+                        args.device,
+                        args.num_iter,
+                    )
 
                 if not args.profile_only:
-                    run_e2e(shape, dtype, channels_last, backward, args.device, args.num_iter)
+                    run_e2e(
+                        shape,
+                        dtype,
+                        channels_last,
+                        backward,
+                        args.device,
+                        args.num_iter,
+                    )
 
 def parse_args():
-    parser = argparse.ArgumentParser(description='OP Benchmark')
-    parser.add_argument('--device', type=str, default='xpu',
-                        help='Device to run on (e.g., "cpu", "cuda", "xpu")')
+    parser = argparse.ArgumentParser(description="OP Benchmark")
+    parser.add_argument(
+        "--device",
+        type=str,
+        default='xpu',
+        help='Device to run on (e.g., "cpu", "cuda", "xpu")'
+    )
     group = parser.add_mutually_exclusive_group()
-    group.add_argument('--profile-only', action='store_true',
-                       help='Only Run profile timing')
-    group.add_argument('--e2e-only', action='store_true',
-                       help='Only Run E2E timing')
-    parser.add_argument('--num-iter', type=int, default=20,
-                        help='Number of iterations')
+    group.add_argument(
+        "--profile-only", action="store_true", help="Only Run profile timing"
+    )
+    group.add_argument("--e2e-only", action="store_true", help="Only Run E2E timing")
+    parser.add_argument("--num-iter", type=int, default=20, help="Number of iterations")
     return parser.parse_args()
 
 if __name__ == "__main__":

test/microbench/indexing.index_fill.py

@@ -69,14 +69,14 @@ def benchmark(args):
 
 def parse_args():
     parser = argparse.ArgumentParser(description='OP Benchmark')
-    parser.add_argument('--device', type=str, default='xpu', 
+    parser.add_argument('--device', type=str, default='xpu',
                         help='Device to run on (e.g., "cpu", "cuda", "xpu")')
     group = parser.add_mutually_exclusive_group()
-    group.add_argument('--profile-only', action='store_true', 
+    group.add_argument('--profile-only', action='store_true',
                        help='Only Run profile timing')
-    group.add_argument('--e2e-only', action='store_true', 
+    group.add_argument('--e2e-only', action='store_true',
                        help='Only Run E2E timing')
-    parser.add_argument('--num-iter', type=int, default=20, 
+    parser.add_argument('--num-iter', type=int, default=20,
                         help='Number of iterations')
     return parser.parse_args()