mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
e0fe5699c4
Summary: Since issue https://github.com/pytorch/pytorch/issues/59859 is resolved. rerun_cmake in build_libtorch should not be hardcoded. build_libtorch is necessary to generate debug version libtorch. Pull Request resolved: https://github.com/pytorch/pytorch/pull/63074 Reviewed By: VitalyFedyunin, seemethere Differential Revision: D30306705 Pulled By: malfet fbshipit-source-id: f2077d334191f4973da0681560937bc8bab730c1
24 lines
1.1 KiB
Python
24 lines
1.1 KiB
Python
import argparse
|
|
from os.path import dirname, abspath
|
|
import sys
|
|
|
|
# By appending pytorch_root to sys.path, this module can import other torch
|
|
# modules even when run as a standalone script. i.e., it's okay either you
|
|
# do `python build_libtorch.py` or `python -m tools.build_libtorch`.
|
|
pytorch_root = dirname(dirname(abspath(__file__)))
|
|
sys.path.append(pytorch_root)
|
|
|
|
from tools.build_pytorch_libs import build_caffe2
|
|
from tools.setup_helpers.cmake import CMake
|
|
|
|
if __name__ == '__main__':
|
|
# Placeholder for future interface. For now just gives a nice -h.
|
|
parser = argparse.ArgumentParser(description='Build libtorch')
|
|
parser.add_argument('--rerun-cmake', action="store_true", help='rerun cmake')
|
|
parser.add_argument('--cmake-only', action="store_true",
|
|
help='Stop once cmake terminates. Leave users a chance to adjust build options')
|
|
options = parser.parse_args()
|
|
|
|
build_caffe2(version=None, cmake_python_library=None, build_python=False,
|
|
rerun_cmake=options.rerun_cmake, cmake_only=options.cmake_only, cmake=CMake())
|