mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
7cae7902a2
Traverses the docs and code to find any broken links Pull Request resolved: https://github.com/pytorch/pytorch/pull/151844 Approved by: https://github.com/huydhn
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
"""
|
|
This is refactored from cmake.py to avoid circular imports issue with env.py,
|
|
which calls get_cmake_cache_variables_from_file
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from typing import IO, Optional, Union
|
|
|
|
|
|
CMakeValue = Optional[Union[bool, str]]
|
|
|
|
|
|
def convert_cmake_value_to_python_value(
|
|
cmake_value: str, cmake_type: str
|
|
) -> CMakeValue:
|
|
r"""Convert a CMake value in a string form to a Python value.
|
|
|
|
Args:
|
|
cmake_value (string): The CMake value in a string form (e.g., "ON", "OFF", "1").
|
|
cmake_type (string): The CMake type of :attr:`cmake_value`.
|
|
|
|
Returns:
|
|
A Python value corresponding to :attr:`cmake_value` with type :attr:`cmake_type`.
|
|
"""
|
|
|
|
cmake_type = cmake_type.upper()
|
|
up_val = cmake_value.upper()
|
|
if cmake_type == "BOOL":
|
|
# https://cmake.org/cmake/help/latest/manual/cmake-generator-expressions.7.html#genex:BOOL
|
|
return not (
|
|
up_val in ("FALSE", "OFF", "N", "NO", "0", "", "NOTFOUND")
|
|
or up_val.endswith("-NOTFOUND")
|
|
)
|
|
elif cmake_type == "FILEPATH":
|
|
if up_val.endswith("-NOTFOUND"):
|
|
return None
|
|
else:
|
|
return cmake_value
|
|
else: # Directly return the cmake_value.
|
|
return cmake_value
|
|
|
|
|
|
def get_cmake_cache_variables_from_file(
|
|
cmake_cache_file: IO[str],
|
|
) -> dict[str, CMakeValue]:
|
|
r"""Gets values in CMakeCache.txt into a dictionary.
|
|
|
|
Args:
|
|
cmake_cache_file: A CMakeCache.txt file object.
|
|
Returns:
|
|
dict: A ``dict`` containing the value of cached CMake variables.
|
|
"""
|
|
|
|
results = {}
|
|
for i, line in enumerate(cmake_cache_file, 1):
|
|
line = line.strip()
|
|
if not line or line.startswith(("#", "//")):
|
|
# Blank or comment line, skip
|
|
continue
|
|
|
|
# Almost any character can be part of variable name and value. As a practical matter, we assume the type must be
|
|
# valid if it were a C variable name. It should match the following kinds of strings:
|
|
#
|
|
# USE_CUDA:BOOL=ON
|
|
# "USE_CUDA":BOOL=ON
|
|
# USE_CUDA=ON
|
|
# USE_CUDA:=ON
|
|
# Intel(R) MKL-DNN_SOURCE_DIR:STATIC=/path/to/pytorch/third_party/ideep/mkl-dnn
|
|
# "OpenMP_COMPILE_RESULT_CXX_openmp:experimental":INTERNAL=FALSE
|
|
matched = re.match(
|
|
r'("?)(.+?)\1(?::\s*([a-zA-Z_-][a-zA-Z0-9_-]*)?)?\s*=\s*(.*)', line
|
|
)
|
|
if matched is None: # Illegal line
|
|
raise ValueError(f"Unexpected line {i} in {repr(cmake_cache_file)}: {line}")
|
|
_, variable, type_, value = matched.groups()
|
|
if type_ is None:
|
|
type_ = ""
|
|
if type_.upper() in ("INTERNAL", "STATIC"):
|
|
# CMake internal variable, do not touch
|
|
continue
|
|
results[variable] = convert_cmake_value_to_python_value(value, type_)
|
|
|
|
return results
|