tests: benchdnn: fix multiple output check logic

oneapi-src · Nov 9, 2023 · 98dc9db · 98dc9db
1 parent 5476ef7
commit 98dc9db
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/benchdnn/graph/graph_bridge.hpp b/tests/benchdnn/graph/graph_bridge.hpp
@@ -239,7 +239,12 @@ void check_correctness(ref_prims_t &ref_prims, size_t op_id, const args_t &args,
             res->state = EXECUTED;
             res->errors = 0;
             cmp.set_norm_validation_mode(true);
-            cmp.compare(mem_fp_abx, mem_dt, prb->attr, res);
+            if (cmp.compare(mem_fp_abx, mem_dt, prb->attr, res) == FAIL) {
+                const std::string norm_check_fail = "Norm check failed, quit!";
+                BENCHDNN_PRINT(
+                        0, "Output arg %d: %s\n", arg, norm_check_fail.c_str());
+                break;
+            }
         }
     }
 }