From e0fe5699c4b7dabd132834b690d6cc2513e0e978 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Fri, 20 Aug 2021 16:28:39 -0700 Subject: [PATCH] enable increment build for build_libtorch (#63074) 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 --- tools/build_libtorch.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/build_libtorch.py b/tools/build_libtorch.py index 800d8eb..c263e50 100644 --- a/tools/build_libtorch.py +++ b/tools/build_libtorch.py @@ -14,7 +14,10 @@ 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=True, cmake_only=False, cmake=CMake()) + rerun_cmake=options.rerun_cmake, cmake_only=options.cmake_only, cmake=CMake()) -- 2.7.4