Skip to content

Commit 6828097

Browse files
authored
[Executorch][BE] Fix error logging with better message (#10339)
ATT Differential Revision: [D73222738](https://our.internmc.facebook.com/intern/diff/D73222738/)
1 parent 77c48f7 commit 6828097

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

extension/llm/custom_ops/op_sdpa_impl.h

+5-1
Original file line numberDiff line numberDiff line change
@@ -605,7 +605,11 @@ void cpu_flash_attention(
605605
*/
606606
ET_CHECK_MSG(attn_mask.value().dim() == 2, "attn_mask must be 2D");
607607
ET_CHECK_MSG(
608-
attn_mask.value().size(0) == qSize, "attn_mask shape mismatch");
608+
attn_mask.value().size(0) == qSize,
609+
"attn_mask shape mismatch"
610+
"attn_mask.size(0)=%zd qSize=%" PRId64,
611+
attn_mask.value().size(0),
612+
qSize);
609613
ET_CHECK_MSG(
610614
attn_mask.value().size(1) == kvSize,
611615
"attn_mask shape mismatch"

0 commit comments

Comments
 (0)