Add comment

sshlyapn · sshlyapn · commit a655f9b004b3 · 2025-03-13T12:26:42.000+04:00
diff --git a/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp b/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp
@@ -23,6 +23,9 @@ static bool is_shape_of_subgraph_root(program_node& node) {
         return true;
     }
 
+    // Allow input_layout to be the root of the shape_of subgraph if it's 'max_context_len'
+    // input of PagedAttention, which can be used as a shape calculation flow source in some
+    // models like Qwen and Qwen2
     if (node.is_type<input_layout>()) {
         const auto& users = node.get_users();
         for (const auto& user : users) {

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,9 @@ static bool is_shape_of_subgraph_root(program_node& node) {`
`23`	`23`	`return true;`
`24`	`24`	`}`
`25`	`25`
	`26`	`+ // Allow input_layout to be the root of the shape_of subgraph if it's 'max_context_len'`
	`27`	`+ // input of PagedAttention, which can be used as a shape calculation flow source in some`
	`28`	`+ // models like Qwen and Qwen2`
`26`	`29`	`if (node.is_type<input_layout>()) {`
`27`	`30`	`const auto& users = node.get_users();`
`28`	`31`	`for (const auto& user : users) {`