initialize is_packed_qkv_

microsoft · Nov 4, 2024 · d49ecb4 · d49ecb4
1 parent 53fd3b0
commit d49ecb4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxruntime/contrib_ops/webgpu/bert/attention_common.h b/onnxruntime/contrib_ops/webgpu/bert/attention_common.h
@@ -57,6 +57,7 @@ struct WebgpuAttentionParameters {
                                                                                               kv_num_heads_(parameters.kv_num_heads),
                                                                                               num_splits_(parameters.num_splits),
                                                                                               rotary_dim_(parameters.rotary_dim),
+                                                                                              is_packed_qkv_(parameters.is_packed_qkv),
                                                                                               is_subsequent_prompt_(parameters.is_subsequent_prompt),
                                                                                               is_first_prompt_(parameters.is_first_prompt),
                                                                                               rotary_interleaved_(parameters.rotary_interleaved),