gahow/aituner
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Prioritize uncovered prefill scheduler candidates

Browse Source
This commit is contained in:
Gahow Wang
2026-06-29 01:30:34 +08:00
parent 36c301c128
commit bfd85793f3
3 changed files with 134 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/aituner/harness.py
View File

@@ -1738,11 +1738,18 @@ def _prefill_scheduler_candidate_actions(
relief = 0.38
if admission_step is not None:
relief += 0.06
score = relief * max(confidence, 0.35) + _information_gain(bottleneck_hypotheses, "runtime") + 0.08
coverage_bonus = 0.0
if quantum_step["direction"] == "seed" or not current_chunked:
coverage_bonus = 0.28
elif quantum_step["target"] is not None or admission_step is not None:
coverage_bonus = 0.14
information_gain = _information_gain(bottleneck_hypotheses, "runtime")
score = relief * max(confidence, 0.35) + information_gain + 0.08 + coverage_bonus
factors = {
"expected_bottleneck_relief": round(relief, 4),
"bottleneck_confidence": round(confidence, 4),
"information_gain": round(_information_gain(bottleneck_hypotheses, "runtime"), 4),
"information_gain": round(information_gain, 4),
"uncovered_scheduler_dimension_bonus": round(coverage_bonus, 4),
"launch_safety": 0.08,
"regression_risk": 0.06 if current_chunked else 0.1,
"prefill_quantum_ratio_current": (