mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a11c1bbdd0
Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/76089 Approved by: https://github.com/albanD
25 lines
812 B
Python
25 lines
812 B
Python
"""NumPy helper.
|
|
|
|
Note: If you plan to add a library detection script like this one, consider it twice. Most library detection should go
|
|
to CMake script. This one is an exception, because Python code can do a much better job due to NumPy's inherent Pythonic
|
|
nature.
|
|
"""
|
|
|
|
from .env import check_negative_env_flag
|
|
|
|
|
|
# Set USE_NUMPY to what the user wants, because even if we fail here, cmake
|
|
# will check for the presence of NumPy again (`cmake/Dependencies.cmake`).
|
|
USE_NUMPY = not check_negative_env_flag("USE_NUMPY")
|
|
NUMPY_INCLUDE_DIR = None
|
|
|
|
if USE_NUMPY:
|
|
try:
|
|
import numpy as np
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
# To reach here, the user must has not disabled NumPy build and the
|
|
# NumPy library is present in the system.
|
|
NUMPY_INCLUDE_DIR = np.get_include()
|