pytorch

1
import argparse
2
import sys
3
from os.path import abspath, dirname
4

5

6
# By appending pytorch_root to sys.path, this module can import other torch
7
# modules even when run as a standalone script. i.e., it's okay either you
8
# do `python build_libtorch.py` or `python -m tools.build_libtorch`.
9
pytorch_root = dirname(dirname(abspath(__file__)))
10
sys.path.append(pytorch_root)
11

12
from tools.build_pytorch_libs import build_caffe2
13
from tools.setup_helpers.cmake import CMake
14

15

16
if __name__ == "__main__":
17
    # Placeholder for future interface. For now just gives a nice -h.
18
    parser = argparse.ArgumentParser(description="Build libtorch")
19
    parser.add_argument("--rerun-cmake", action="store_true", help="rerun cmake")
20
    parser.add_argument(
21
        "--cmake-only",
22
        action="store_true",
23
        help="Stop once cmake terminates. Leave users a chance to adjust build options",
24
    )
25
    options = parser.parse_args()
26

27
    build_caffe2(
28
        version=None,
29
        cmake_python_library=None,
30
        build_python=False,
31
        rerun_cmake=options.rerun_cmake,
32
        cmake_only=options.cmake_only,
33
        cmake=CMake(),
34
    )
35