We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 77c48f7 commit 6828097Copy full SHA for 6828097
extension/llm/custom_ops/op_sdpa_impl.h
@@ -605,7 +605,11 @@ void cpu_flash_attention(
605
*/
606
ET_CHECK_MSG(attn_mask.value().dim() == 2, "attn_mask must be 2D");
607
ET_CHECK_MSG(
608
- attn_mask.value().size(0) == qSize, "attn_mask shape mismatch");
+ 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);
613
614
attn_mask.value().size(1) == kvSize,
615
"attn_mask shape mismatch"
0 commit comments