saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/scripts
History
Zhengxu Chen 3ef2dfc1ba [export] Implement cpp deserializer. (#136398) 
Differential Revision: D63206258

This diff introduces a mechanism to generate a json-compatible deserializer in cpp using nlohmann json (already being used by AOTI).

Why we need this? Because there will be a lot of cases where people don't want to use Python to load the graph (e.g. cpp runtime), and instead they can use this header to deserialize the JSON graph.

Every time we call update_schema.py to update the schema, the header will be auto generated and included into the source files.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136398
Approved by: https://github.com/angelayi
2024-11-14 16:34:59 +00:00
..
analysis [BE][Easy][1/19] enforce style for empty lines in import segments (#129752) 2024-07-16 00:42:56 +00:00
compile_tests change name of dynamo CI chard to dynamo_wrapped (#138233) 2024-10-28 21:42:33 +00:00
export [export] Implement cpp deserializer. (#136398) 2024-11-14 16:34:59 +00:00
fbcode-dev-setup
jit [BE][Easy][1/19] enforce style for empty lines in import segments (#129752) 2024-07-16 00:42:56 +00:00
onnx
release Fix script name in the comments (#135507) 2024-09-13 19:59:47 +00:00
release_notes [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
add_apache_header.sh
apache_header.txt
apache_python.txt
build_android.sh [BE][CMake] Use FindPython module (#124613) 2024-05-29 13:17:35 +00:00
build_android_gradle.sh
build_host_protoc.sh
build_ios.sh [BE][CMake] Use FindPython module (#124613) 2024-05-29 13:17:35 +00:00
build_local.sh
build_mobile.sh [BE][CMake] Use FindPython module (#124613) 2024-05-29 13:17:35 +00:00
build_pytorch_android.sh
build_raspbian.sh
build_tegra_x1.sh [1/2] Remove caffe2 db and distributed from build system (#125092) 2024-05-04 06:48:46 +00:00
build_tizen.sh [1/2] Remove caffe2 db and distributed from build system (#125092) 2024-05-04 06:48:46 +00:00
build_windows.bat [CMake] Remove BUILDING_WITH_TORCH_LIBS (#134434) 2024-08-27 01:48:21 +00:00
diagnose_protobuf.py [BE][Easy][1/19] enforce style for empty lines in import segments (#129752) 2024-07-16 00:42:56 +00:00
get_python_cmake_flags.py [BE][Easy][1/19] enforce style for empty lines in import segments (#129752) 2024-07-16 00:42:56 +00:00
install_triton_wheel.sh [XPU] Set make triton install pre-built whl by default (#130313) 2024-08-23 07:36:34 +00:00
proto.ps1
read_conda_versions.sh
README.md
remove_apache_header.sh
temp.sh
xcode_build.rb

README.md

This directory contains the useful tools.

build_android.sh

This script is to build PyTorch/Caffe2 library for Android. Take the following steps to start the build:

  • set ANDROID_NDK to the location of ndk
export ANDROID_NDK=YOUR_NDK_PATH
  • run build_android.sh
#in your PyTorch root directory
bash scripts/build_android.sh

If succeeded, the libraries and headers would be generated to build_android/install directory. You can then copy these files from build_android/install to your Android project for further usage.

You can also override the cmake flags via command line, e.g., following command will also compile the executable binary files:

bash scripts/build_android.sh -DBUILD_BINARY=ON

build_ios.sh

This script is to build PyTorch/Caffe2 library for iOS, and can only be performed on macOS. Take the following steps to start the build:

  • Install Xcode from App Store, and configure "Command Line Tools" properly on Xcode.
  • Install the dependencies:
brew install cmake automake libtool
  • run build_ios.sh
#in your PyTorch root directory
bash scripts/build_ios.sh

If succeeded, the libraries and headers would be generated to build_ios/install directory. You can then copy these files to your Xcode project for further usage.