mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
21d7532dfe
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30417 Differential Revision: D18716502 Pulled By: albanD fbshipit-source-id: 0b1b86f882e0e24cb6845e4a44708048e7e3b4a8
24 lines
812 B
Python
24 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()
|