From: Yi Zhang Date: Fri, 20 Aug 2021 23:28:39 +0000 (-0700) Subject: enable increment build for build_libtorch (#63074) X-Git-Tag: accepted/tizen/8.0/unified/20231005.095509~827 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e0fe5699c4b7dabd132834b690d6cc2513e0e978;p=platform%2Fupstream%2Fpytorch.git 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 --- 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())