[webgpu] Fix graph capture broken (microsoft#26383)

qjia7 · web-flow · commit 949c0de7d0ee · 2025-10-23T00:33:51.000+08:00
microsoft#26297 deleted `seqlen_k` by accident which broke the graph capture in phi4. This PR brings it back.
diff --git a/onnxruntime/contrib_ops/webgpu/bert/group_query_attention.cc b/onnxruntime/contrib_ops/webgpu/bert/group_query_attention.cc
@@ -287,7 +287,7 @@ Status GroupQueryAttention::ComputeInternal(onnxruntime::webgpu::ComputeContext&
       !use_sliding_window &&
       CanApplyFlashAttention(attention_bias, present_key, present_value, parameters, context)) {
     return ApplyFlashAttention(query, key, value, attention_bias, output, past_key, present_key, past_value,
-                               present_value, parameters, context);
+                               present_value, parameters, context, seqlen_k);
   }
 
   TensorShapeVector q_new_dims({parameters.batch_size_, parameters.num_heads_,

Original file line number	Diff line number	Diff line change
`@@ -287,7 +287,7 @@ Status GroupQueryAttention::ComputeInternal(onnxruntime::webgpu::ComputeContext&`
`287`	`287`	`!use_sliding_window &&`
`288`	`288`	`CanApplyFlashAttention(attention_bias, present_key, present_value, parameters, context)) {`
`289`	`289`	`return ApplyFlashAttention(query, key, value, attention_bias, output, past_key, present_key, past_value,`
`290`		`- present_value, parameters, context);`
	`290`	`+ present_value, parameters, context, seqlen_k);`
`291`	`291`	`}`
`292`	`292`
`293`	`293`	`TensorShapeVector q_new_dims({parameters.batch_size_, parameters.num_heads_,`