Skip to content

Commit f033621

Browse files
khluucomaniac
authored andcommitted
[CI/Build] Fix machete generated kernel files ordering (vllm-project#8976)
Signed-off-by: kevin <[email protected]> Co-authored-by: Cody Yu <[email protected]> Signed-off-by: Sumit Dubey <[email protected]>
1 parent 86127d4 commit f033621

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

csrc/quantization/machete/generate.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -457,7 +457,13 @@ def generate():
457457
)),
458458
]
459459

460-
schedules = list(set([x[1] for x in default_heuristic]))
460+
# Do not use schedules = list(set(...)) because we need to make sure
461+
# the output list is deterministic; otherwise the generated kernel file
462+
# will be non-deterministic and causes ccache miss.
463+
schedules = []
464+
for _, schedule_config in default_heuristic:
465+
if schedule_config not in schedules:
466+
schedules.append(schedule_config)
461467

462468
impl_configs = []
463469

0 commit comments

Comments
 (0)