From e07c4a39b59575fb5a8c4115b6a7e54407cd567a Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Wed, 11 Dec 2024 22:45:25 -0800
Subject: [PATCH] packaging: add ninja to dependencies (#654)

otherwise aot compilation will be too slow.
---
 custom_backend.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/custom_backend.py b/custom_backend.py
index 8e5ce2d8..b95f7004 100644
--- 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):