[CI/Build] Fix machete generated kernel files ordering (vllm-project#…

…8976) Signed-off-by: kevin <[email protected]> Co-authored-by: Cody Yu <[email protected]>
dtrifiro · Oct 3, 2024 · 5bd6b70 · 5bd6b70
1 parent 9adefcf
commit 5bd6b70
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/csrc/quantization/machete/generate.py b/csrc/quantization/machete/generate.py
@@ -457,7 +457,13 @@ def generate():
             )),
     ]
 
-    schedules = list(set([x[1] for x in default_heuristic]))
+    # Do not use schedules = list(set(...)) because we need to make sure
+    # the output list is deterministic; otherwise the generated kernel file
+    # will be non-deterministic and causes ccache miss.
+    schedules = []
+    for _, schedule_config in default_heuristic:
+        if schedule_config not in schedules:
+            schedules.append(schedule_config)
 
     impl_configs = []