Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 1 | import argparse |
Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 2 | import sys |
Huy Do | 347b036 | 2022-07-13 07:59:20 +0000 | [diff] [blame] | 3 | from os.path import abspath, dirname |
Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 4 | |
Derek Kim | ec8b1c9 | 2019-01-16 23:52:37 -0800 | [diff] [blame] | 5 | # By appending pytorch_root to sys.path, this module can import other torch |
| 6 | # modules even when run as a standalone script. i.e., it's okay either you |
| 7 | # do `python build_libtorch.py` or `python -m tools.build_libtorch`. |
| 8 | pytorch_root = dirname(dirname(abspath(__file__))) |
| 9 | sys.path.append(pytorch_root) |
| 10 | |
Zachary DeVito | 9477a5d | 2019-01-25 15:57:09 -0800 | [diff] [blame] | 11 | from tools.build_pytorch_libs import build_caffe2 |
Hong Xu | b811b6d | 2019-06-14 08:10:18 -0700 | [diff] [blame] | 12 | from tools.setup_helpers.cmake import CMake |
Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 13 | |
Edward Z. Yang | a11c1bb | 2022-04-20 08:51:02 -0400 | [diff] [blame] | 14 | if __name__ == "__main__": |
Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 15 | # Placeholder for future interface. For now just gives a nice -h. |
Edward Z. Yang | a11c1bb | 2022-04-20 08:51:02 -0400 | [diff] [blame] | 16 | parser = argparse.ArgumentParser(description="Build libtorch") |
| 17 | parser.add_argument("--rerun-cmake", action="store_true", help="rerun cmake") |
| 18 | parser.add_argument( |
| 19 | "--cmake-only", |
| 20 | action="store_true", |
| 21 | help="Stop once cmake terminates. Leave users a chance to adjust build options", |
| 22 | ) |
Peter Goldsborough | 130d55a | 2018-09-12 16:40:30 -0700 | [diff] [blame] | 23 | options = parser.parse_args() |
Anders Papitto | 130881f | 2018-08-16 07:57:00 -0700 | [diff] [blame] | 24 | |
Edward Z. Yang | a11c1bb | 2022-04-20 08:51:02 -0400 | [diff] [blame] | 25 | build_caffe2( |
| 26 | version=None, |
| 27 | cmake_python_library=None, |
| 28 | build_python=False, |
| 29 | rerun_cmake=options.rerun_cmake, |
| 30 | cmake_only=options.cmake_only, |
| 31 | cmake=CMake(), |
| 32 | ) |