-
Notifications
You must be signed in to change notification settings - Fork 2.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[HETERO] Support splitting new graph pattern for pipeline parallel and correct the number of submodels #25224
[HETERO] Support splitting new graph pattern for pipeline parallel and correct the number of submodels #25224
Conversation
aa1e1c0
to
09c2109
Compare
67c96a4
to
7d1b5bc
Compare
bool is_shapeof = ov::is_type<op::util::ShapeOfBase>(op); | ||
if (((fused_model_op_map.find(name) != fused_model_op_map.end()) || is_shapeof) && | ||
supported.count(name)) { | ||
if ((!supported.count(fused_model_op_map[name]) || is_shapeof) && |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
seems this is a special case(shape of) of codes line 401 to 408?
maybe we can consider optimize in future?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It is a problem that occurs when recursively looping through the entire graph, move shape of case to L401-L408 can not fix this issues, we will try the simpler way to optimize this API
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please create a ticket to follow up if there is a way to make this solution more common.
…d correct the number of submodels (openvinotoolkit#25224) ### Details: - Fix qwen1.5-14b-chat with HETERO pipeline parallelism Add supported to patten: ``` ReadValue->Gather->Concat |------>ShapeOf(fused on other different affinity node) ->.... ``` - Correct the value of HETERO_NUMBER_OF_SUBMODELS by subtracting the number of independent submodels to reduce confusion ### Tickets: - *ticket-id*
Details:
Add supported to patten:
Tickets: