format

Author: yucai-intel

test/microbench/grid_sampler.grid_sampler_2d.py

@@ -2,8 +2,8 @@
 from torch.profiler import profile, ProfilerActivity
 
 shape_list = [
-    (6, 5, 7, 3),
-    (4, 32, 128, 128),
+    # (6, 5, 7, 3),
+    # (4, 32, 128, 128),
     (16, 128, 512, 512),
 ]
 
@@ -12,8 +12,8 @@
 
 for shape in shape_list:
     for dtype in [torch.bfloat16, torch.float16, torch.float32]:
-        for mode in ["bilinear", "nearest", "bicubic"]:
-            for padding_mode in ["zeros", "border", "reflection"]:
+        for padding_mode in ["zeros", "border", "reflection"]:
+            for mode in ["bilinear", "bicubic","nearest"]:
                 for align_corners in [True, False]:
                     N, C, H, W = shape