packaging: add ninja to dependencies (#654)

otherwise aot compilation will be too slow.
flashinfer-ai · Dec 12, 2024 · e07c4a3 · e07c4a3
1 parent 8e3d258
commit e07c4a3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_backend.py b/custom_backend.py
@@ -6,7 +6,11 @@
 
 
 def _get_requires_for_build():
-    return ["torch"] if os.environ.get("FLASHINFER_ENABLE_AOT", "0") == "1" else []
+    return (
+        ["torch", "ninja"]
+        if os.environ.get("FLASHINFER_ENABLE_AOT", "0") == "1"
+        else ["ninja"]
+    )
 
 
 def get_requires_for_build_wheel(config_settings=None):