From e3ae18addb508aac1f49681d686b0adac05a1d90 Mon Sep 17 00:00:00 2001
From: NULLPlayer <zhoushun.jackson@gmail.com>
Date: Tue, 28 Jan 2025 12:20:12 +0800
Subject: [PATCH 1/2] build disable ninja

---
 setup.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/setup.py b/setup.py
index 3df847dd5..fd15c13b3 100644
--- a/setup.py
+++ b/setup.py
@@ -16,6 +16,16 @@ try:
         from torch_mlu.utils.cpp_extension import BuildExtension
         EXT_TYPE = 'pytorch'
     else:
+        # build disable ninja
+        if os.getenv("MMCV_NO_NINJA", "0") == "1":
+        
+            class BuildExtension(torch.utils.cpp_extension.BuildExtension):
+                def __init__(self, *args, **kwargs):
+                    super().__init__(use_ninja=False, *args, **kwargs)
+        
+        else:
+            BuildExtension = torch.utils.cpp_extension.BuildExtension
+
         from torch.utils.cpp_extension import BuildExtension
         EXT_TYPE = 'pytorch'
     cmd_class = {'build_ext': BuildExtension}

From 402ccb3cf91711f2960e46ecbe08b510a6e48d8d Mon Sep 17 00:00:00 2001
From: NULLPlayer <zhoushun.jackson@gmail.com>
Date: Tue, 28 Jan 2025 12:29:16 +0800
Subject: [PATCH 2/2] build disable ninja

---
 setup.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/setup.py b/setup.py
index fd15c13b3..ced20fd15 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,6 @@ try:
         else:
             BuildExtension = torch.utils.cpp_extension.BuildExtension
 
-        from torch.utils.cpp_extension import BuildExtension
         EXT_TYPE = 'pytorch'
     cmd_class = {'build_ext': BuildExtension}
 except ModuleNotFoundError: