[tuner]: format the code

Signed-off-by: Bangtian Liu <[email protected]>
nod-ai · Dec 9, 2024 · 89e7b1a · 89e7b1a
1 parent aa4f2d5
commit 89e7b1a
Showing 1 changed file with 2 additions and 18 deletions.
diff --git a/tuner/tuner/dispatch_parser_test.py b/tuner/tuner/dispatch_parser_test.py
@@ -80,24 +80,8 @@ def test_get_conv_tile_sizes(tuner_ctx: common.TunerContext) -> None:
         gpu_pipeline_options=iree_gpu.PipelineOptionsAttr.get(),
         waves_per_eu=1,
     )
-    assert config.lowering_config.workgroup_tile_sizes == [
-        1,
-        1,
-        464,
-        320,
-        1,
-        1,
-        0,
-    ]
-    assert config.lowering_config.reduction_tile_sizes == [
-        0,
-        0,
-        0,
-        0,
-        0,
-        0,
-        16,
-    ]
+    assert config.lowering_config.workgroup_tile_sizes == [1, 1, 464, 320, 1, 1, 0]
+    assert config.lowering_config.reduction_tile_sizes == [0, 0, 0, 0, 0, 0, 16]
 
 
 def test_get_contract_tile_sizes(tuner_ctx: common.TunerContext) -> None: