diff --git a/setup.py b/setup.py index c3e1f500d..ffc24548b 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,16 @@ try: from torch_musa.utils.musa_extension import BuildExtension EXT_TYPE = 'pytorch' else: - from torch.utils.cpp_extension import BuildExtension + # 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 + EXT_TYPE = 'pytorch' cmd_class = {'build_ext': BuildExtension} except ModuleNotFoundError: