Skip to content

Commit bca73e5

Browse files
author
BrianPark314
committed
feat: add prefix aware picker
Signed-off-by: BrianPark314 <[email protected]>
1 parent 53500d6 commit bca73e5

File tree

3 files changed

+3
-1
lines changed

3 files changed

+3
-1
lines changed

src/gateway_inference_extension/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ RUN git clone https://github.com/kubernetes-sigs/gateway-api-inference-extension
2525
git apply scheduler.patch && \
2626
cd ../../../.. && \
2727
cp /src/roundrobin_picker.go gateway-api-inference-extension/pkg/epp/scheduling/plugins/picker/roundrobin_picker.go && \
28+
cp /src/prefix_aware_picker.go gateway-api-inference-extension/pkg/epp/scheduling/plugins/picker/prefix_aware_picker.go && \
2829
mkdir -p /src/pkg/ && \
2930
cp -r gateway-api-inference-extension/pkg/epp/ /src/pkg/epp && \
3031
cp gateway-api-inference-extension/go.mod /src && \

src/gateway_inference_extension/prefix_aware_picker.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,4 +149,4 @@ func (t *hashTrie) longestPrefixMatch(
149149
}
150150
}
151151
return res
152-
}
152+
}

src/gateway_inference_extension/scheduler.patch

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ index b484cde..c7688a8 100644
1616
scorers: map[plugins.Scorer]int{},
1717
- picker: &picker.RandomPicker{},
1818
+ picker: &picker.RoundRobinPicker{},
19+
+ picker: &picker.PrefixAwarePicker{},
1920
postSchedulePlugins: []plugins.PostSchedule{},
2021
}
2122

0 commit comments

Comments
 (0)