diff --git a/examples/10_bmg_grouped_gemm_mixed_dtype/bmg_grouped_gemm_mixed_dtype_runner.hpp b/examples/10_bmg_grouped_gemm_mixed_dtype/bmg_grouped_gemm_mixed_dtype_runner.hpp index 71ef822217..592e816f2c 100644 --- a/examples/10_bmg_grouped_gemm_mixed_dtype/bmg_grouped_gemm_mixed_dtype_runner.hpp +++ b/examples/10_bmg_grouped_gemm_mixed_dtype/bmg_grouped_gemm_mixed_dtype_runner.hpp @@ -498,7 +498,7 @@ struct ExampleRunner { ElementOutput const epsilon(1e-2f); ElementOutput const non_zero_floor(1e-4f); - bool passed = false; + bool passed = true; for(int i = 0; i < options.groups; i++) { Shape problem_size = append<4>(options.problem_sizes_host[i], 1); @@ -534,7 +534,9 @@ struct ExampleRunner { compat::wait(); // compare_reference - passed |= cutlass::reference::device::BlockCompareRelativelyEqual(block_ref_D.get(), block_D.get() + offset_D[i], block_ref_D.size(), epsilon, non_zero_floor); + passed &= cutlass::reference::device::BlockCompareRelativelyEqual(block_ref_D.get(), block_D.get() + offset_D[i], block_ref_D.size(), epsilon, non_zero_floor); + if(!passed) + break; compat::wait(); }