Skip to content

Commit 73ecf23

Browse files
authored
[Executorch][BE] Fix error logging with better message
Differential Revision: D73222738 Pull Request resolved: #10283
1 parent 0d00ca7 commit 73ecf23

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

extension/llm/custom_ops/op_sdpa_impl.h

Lines changed: 5 additions & 1 deletion
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)