pytorch/scripts
BowenBao 68034197e8 [ONNX] Support gelu for fp16 export (#50487) (#50911)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50911

Need to replace dtype of export created scalars from float to double. (In torch implicit conversion logic, python numbers are double)

Test case skipped in CI due to that current CI job env does not have CUDA support.

Test Plan: Imported from OSS

Reviewed By: pbelevich

Differential Revision: D26050889

Pulled By: SplitInfinity

fbshipit-source-id: 1fdde23a68d4793e6b9a82840acc213e5c3aa760
2021-01-27 17:49:02 -08:00
..
appveyor
fbcode-dev-setup
model_zoo Drop unused imports from scripts (#49956) 2021-01-04 16:08:28 -08:00
onnx [ONNX] Support gelu for fp16 export (#50487) (#50911) 2021-01-27 17:49:02 -08:00
release ci: Add anaconda pruning to CI pipeline (#44651) 2020-09-15 10:51:05 -07:00
release_notes Convert string => raw strings so char classes can be represented in Python regex (#50239) 2021-01-08 11:17:17 -08:00
add_apache_header.sh
apache_header.txt
apache_python.txt
build_android.sh
build_host_protoc.sh
build_ios.sh [CI][IOS] Add a arm64 ios job for Metal (#46646) 2020-10-22 16:54:46 -07:00
build_local.sh
build_mobile.sh
build_pytorch_android.sh
build_raspbian.sh
build_tegra_x1.sh
build_tizen.sh
build_windows.bat
diagnose_protobuf.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
get_python_cmake_flags.py Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
proto.ps1
read_conda_versions.sh
README.md
remove_apache_header.sh
temp.sh
xcode_build.rb [iOS][CI] Upgrade xcode version to 12.0 (#45677) 2020-10-01 16:53:18 -07:00

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.