Skip to content

Consolidate PrefillFilter and DecodeFilter to one class#161

Closed
mayabar wants to merge 5 commits intollm-d:mainfrom
mayabar:pd-filters-consolidation
Closed

Consolidate PrefillFilter and DecodeFilter to one class#161
mayabar wants to merge 5 commits intollm-d:mainfrom
mayabar:pd-filters-consolidation

Conversation

@mayabar
Copy link
Contributor

@mayabar mayabar commented Jun 3, 2025

Create a new class - RoleFilter
Role filter is initialized with list of roles and filters out all pods that are marked with role that is not in the list.
Used for Prefill and Decode filtering

Signed-off-by: Maya Barnea <mayab@il.ibm.com>
@mayabar mayabar requested review from elevran and shmuelk June 3, 2025 06:11
Signed-off-by: Maya Barnea <mayab@il.ibm.com>
@mayabar mayabar requested a review from shmuelk June 3, 2025 08:18
mayabar added 2 commits June 3, 2025 12:02
fix typo

Signed-off-by: Maya Barnea <mayab@il.ibm.com>
Signed-off-by: Maya Barnea <mayab@il.ibm.com>
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can this leverage by_label.go as the base?

ds,
scheduler.generateSchedulerConfig(ctx, schedCfg.PrefillSchedulerPlugins,
&filter.PrefillFilter{}),
filter.NewPrefillFilter()),
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: seems like we could have remove the NewPrefillFilter and NewDecodeFilter functions and just use here:

filter.NewRoleBasedFilter("prefill-filter", filter.RolePrefill)

(same for decode).

@kfirtoledo
Copy link
Collaborator

kfirtoledo commented Jun 16, 2025

@mayabar - What is the status of this PR?

@mayabar
Copy link
Contributor Author

mayabar commented Jun 17, 2025

Closed because the issue was fixed by #188

@mayabar mayabar closed this Jun 17, 2025
@mayabar mayabar deleted the pd-filters-consolidation branch June 17, 2025 12:20
Jooho pushed a commit to Jooho/llm-d-inference-scheduler that referenced this pull request Sep 30, 2025
Signed-off-by: konflux-internal-p02 <170854209+konflux-internal-p02[bot]@users.noreply.github.com>
Co-authored-by: konflux-internal-p02[bot] <170854209+konflux-internal-p02[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants