mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
240d62fbaa
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/21417 Reviewed By: ezyang Differential Revision: D15694357 Pulled By: fmassa fbshipit-source-id: bc1bda23349ba4531f19619fa4adecb846225c20
20 lines
558 B
Python
20 lines
558 B
Python
"NumPy helper."
|
|
|
|
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()
|