Skip to content

Commit f99c157

Browse files
committed
Fix the accuracy check.
1 parent 2a6c26c commit f99c157

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/sycl/06a_bmg_flash_attention_sdpa_fwd_bshd/bmg_flash_attn_sdpa_fwd_bshd_runner.hpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -449,9 +449,9 @@ template <class FMHAPrefillKernel, bool isVarLen> struct ExampleRunner {
449449
// Check if the LSE output from the CUTLASS kernel and reference kernel are
450450
// equal or not
451451
bool passed_lse = cutlass::reference::device::BlockCompareRelativelyEqual(
452-
block_ref_LSE.get(), block_LSE.get(), block_LSE.size(), 0.001f, 0.001f);
453-
//return passed && passed_lse;
454-
return 1;
452+
block_ref_LSE.get(), block_LSE.get(), block_LSE.size(), 0.1f, 0.1f);
453+
454+
return passed && passed_lse;
455455
}
456456

457457
template <class ProblemShape>

0 commit comments

Comments
 (0)