Update e2e test with rhs = indentity_mat * 2

nod-ai · Oct 9, 2024 · 4b2897c · 4b2897c
1 parent 6b9b478
commit 4b2897c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/build_tools/ci/cpu_comparison/run.py b/build_tools/ci/cpu_comparison/run.py
@@ -763,14 +763,15 @@ def run(self, config):
         identity_mat = np.eye(32, dtype=np.float32)
         ones = np.ones(32 * 32, dtype=np.float32).reshape([32, 32])
         lhs = ones * 192
+        rhs = identity_mat * 2
         input_args = generate_inputs(
-            template_name, output_dir, 1, {1: lhs, 2: identity_mat}
+            template_name, output_dir, 1, {1: lhs, 2: rhs}
         )
         aie_vs_baseline(
             config,
             template_name,
             input_args,
-            lhs,  # exected output
+            lhs * 2,  # exected output
             use_ukernel=False,
             tile_pipeline="pack-peel",
             lower_to_aie_pipeline="objectFifo",