Unverified Commit f04f549b authored by 玩火's avatar 玩火 Committed by GitHub
Browse files

Fix is_ninja_available() (#23752)

* Fix is_ninja_available()

search ninja using subprocess instead of importlib.

* Fix style

* Fix doc

* Fix style
parent 3416bba7
...@@ -19,6 +19,7 @@ import importlib.util ...@@ -19,6 +19,7 @@ import importlib.util
import json import json
import os import os
import shutil import shutil
import subprocess
import sys import sys
import warnings import warnings
from collections import OrderedDict from collections import OrderedDict
...@@ -94,7 +95,6 @@ _kenlm_available = _is_package_available("kenlm") ...@@ -94,7 +95,6 @@ _kenlm_available = _is_package_available("kenlm")
_keras_nlp_available = _is_package_available("keras_nlp") _keras_nlp_available = _is_package_available("keras_nlp")
_librosa_available = _is_package_available("librosa") _librosa_available = _is_package_available("librosa")
_natten_available = _is_package_available("natten") _natten_available = _is_package_available("natten")
_ninja_available = _is_package_available("ninja")
_onnx_available = _is_package_available("onnx") _onnx_available = _is_package_available("onnx")
_openai_available = _is_package_available("openai") _openai_available = _is_package_available("openai")
_optimum_available = _is_package_available("optimum") _optimum_available = _is_package_available("optimum")
...@@ -449,7 +449,16 @@ def is_apex_available(): ...@@ -449,7 +449,16 @@ def is_apex_available():
def is_ninja_available(): def is_ninja_available():
return _ninja_available r"""
Code comes from *torch.utils.cpp_extension.is_ninja_available()*. Returns `True` if the
[ninja](https://ninja-build.org/) build system is available on the system, `False` otherwise.
"""
try:
subprocess.check_output("ninja --version".split())
except Exception:
return False
else:
return True
def is_ipex_available(): def is_ipex_available():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment