Restore AIR tests (#710)

nod-ai · Aug 27, 2024 · a79df28 · a79df28
1 parent c2b9f58
commit a79df28
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/build_tools/ci/cpu_comparison/run_test.py b/build_tools/ci/cpu_comparison/run_test.py
@@ -618,24 +618,24 @@ def run(self, config):
                 function_name="three_$mm$",
             )
 
-            # Test(s) of the form matmul(A,B) where A:MxK, B:KxN
-            test_name = output_dir / "test_from_template.mlir"
-            template_name = matmul_template_dir / "matmul_MxK_KxN.mlir"
-            generate_matmul_test(test_name, template_name, 32, 32, 64, "bf16", "f32")
-            aie_vs_llvm_cpu(config, test_name)
-
-            # Test(s) of the form matmul(A,B) + C where A:MxK, B:KxN, C:N
-            test_name = output_dir / "test_from_template_bias_N.mlir"
-            template_name = matmul_template_dir / "matmul_bias_MxK_KxN_N.mlir"
-            generate_matmul_test(
-                test_name, template_name, 1024, 1024, 512, "bf16", "f32"
-            )
-            aie_vs_llvm_cpu(
-                config, test_name, tile_pipeline="pack-peel", use_ukernel=True
-            )
-            aie_vs_llvm_cpu(
-                config, test_name, tile_pipeline="pack-peel", use_ukernel=False
-            )
+        # Test(s) of the form matmul(A,B) where A:MxK, B:KxN
+        test_name = output_dir / "test_from_template.mlir"
+        template_name = matmul_template_dir / "matmul_MxK_KxN.mlir"
+        generate_matmul_test(test_name, template_name, 32, 32, 64, "bf16", "f32")
+        aie_vs_llvm_cpu(config, test_name)
+
+        # Test(s) of the form matmul(A,B) + C where A:MxK, B:KxN, C:N
+        test_name = output_dir / "test_from_template_bias_N.mlir"
+        template_name = matmul_template_dir / "matmul_bias_MxK_KxN_N.mlir"
+        generate_matmul_test(
+            test_name, template_name, 1024, 1024, 512, "bf16", "f32"
+        )
+        aie_vs_llvm_cpu(
+            config, test_name, tile_pipeline="pack-peel", use_ukernel=True
+        )
+        aie_vs_llvm_cpu(
+            config, test_name, tile_pipeline="pack-peel", use_ukernel=False
+        )
 
 
 class SmokeSet(TestSet):