frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Run Black on all of tools/

Browse Source 
Signed-off-by: Edward Z. Yang <ezyangfb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76089

Approved by: https://github.com/albanD
This commit is contained in:
Edward Z. Yang
2022-04-20 08:51:02 -04:00
committed by PyTorch MergeBot
parent ae864d4fb9
commit a11c1bbdd0
79 changed files with 6183 additions and 3745 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tools/build_libtorch.py
View File

@ -11,13 +11,22 @@ sys.path.append(pytorch_root)
from tools.build_pytorch_libs import build_caffe2
from tools.setup_helpers.cmake import CMake
if __name__ == '__main__':
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')
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())
build_caffe2(
version=None,
cmake_python_library=None,
build_python=False,
rerun_cmake=options.rerun_cmake,
cmake_only=options.cmake_only,
cmake=CMake(),
)