-
Notifications
You must be signed in to change notification settings - Fork 8.9k
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
YARN-11610. [Federation] Add WeightedHomePolicyManager. #6256
Conversation
💔 -1 overall
This message was automatically generated. |
🎊 +1 overall
This message was automatically generated. |
@goiri Can you help review this PR? Thank you very much! |
🎊 +1 overall
This message was automatically generated. |
@goiri Can you help review this PR? Thank you very much! |
routerWeights.put(new SubClusterIdInfo(sc1), 0.2f); | ||
policyInfo.setRouterPolicyWeights(routerWeights); | ||
|
||
((WeightedHomePolicyManager) wfp).setWeightedPolicyInfo(policyInfo); |
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.
Can we declare wfp as WeightedHomePolicyManager directly to avoid the casting around?
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.
Thank you very much for your help in reviewing the code! I will improve this part of the code.
🎊 +1 overall
This message was automatically generated. |
🎊 +1 overall
This message was automatically generated. |
@goiri Thank you very much for your help in reviewing the code! I will continue to follow up on YARN-11614. |
… Contributed by Shilun Fan. Reviewed-by: Inigo Goiri <[email protected]> Signed-off-by: Shilun Fan <[email protected]>
Description of PR
JIRA: YARN-11610. [Federation] Add WeightedHomePolicyManager.
The current version of YARN Federation supports cross-cluster allocation of containers, but this feature is not yet fully mature. It will be a key focus for iteration in our next version.
In our current implementation of YARN Federation, after selecting the HomeSubCluster, tasks are executed within that cluster and all containers are allocated within it. We can achieve this using the WeightedLocalityPolicyManager, but it requires extensive configuration, especially for queue amrmWeight.
To simplify usage, we have created a new PolicyManager that selects sub-clusters based on queue weights and ensures tasks are executed entirely within the chosen sub-cluster. This eliminates the need for complex configuration and provides a more user-friendly experience when using YARN Federation.
How was this patch tested?
For code changes:
LICENSE
,LICENSE-binary
,NOTICE-binary
files?