Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65075
Need to drop one line at - https://github.com/pytorch/builder/blob/master/conda/pytorch-nightly/meta.yaml#L65
ghstack-source-id:
138324213
Test Plan:
- Check the iOS nightly builds
- `pod install LibTorch-Lite-Nightly`
Reviewed By: hanton
Differential Revision:
D30912269
fbshipit-source-id:
b07679b75ecf89beae2975c37cf17d2449a3304f
props_dict["ios_platform"] = ios_definitions.get_platform(self.variant)
props_dict["name"] = self.gen_job_name()
props_dict["use_metal"] = miniutils.quote(str(int(True)))
+ props_dict["use_coreml"] = miniutils.quote(str(int(True)))
template_name = "_".join([
"binary",
lite_interpreter:
type: string
default: "1"
+ use_coreml:
+ type: string
+ default: "0"
environment:
BUILD_ENVIRONMENT: << parameters.build_environment >>
IOS_ARCH: << parameters.ios_arch >>
SELECTED_OP_LIST: << parameters.op_list >>
USE_PYTORCH_METAL: << parameters.use_metal >>
BUILD_LITE_INTERPRETER: << parameters.lite_interpreter >>
+ USE_COREML_DELEGATE: << parameters.use_coreml >>
pytorch_windows_params: &pytorch_windows_params
parameters:
ios_arch: x86_64
ios_platform: SIMULATOR
name: pytorch_ios_12_5_1_nightly_x86_64_build
+ use_coreml: "1"
use_metal: "1"
- binary_ios_build:
build_environment: libtorch-ios-12.5.1-nightly-arm64-build
ios_arch: arm64
ios_platform: OS
name: pytorch_ios_12_5_1_nightly_arm64_build
+ use_coreml: "1"
use_metal: "1"
- binary_ios_upload:
build_environment: libtorch-ios-12.5.1-nightly-binary-build-upload
echo "IOS_ARCH: ${IOS_ARCH}"
echo "IOS_PLATFORM: ${IOS_PLATFORM}"
echo "USE_PYTORCH_METAL: ${USE_PYTORCH_METAL}"
+echo "USE_COREML_DELEGATE: ${USE_COREML_DELEGATE}"
export IOS_ARCH=${IOS_ARCH}
export IOS_PLATFORM=${IOS_PLATFORM}
export USE_PYTORCH_METAL=${USE_PYTORCH_METAL}
+export USE_COREML_DELEGATE=${USE_COREML_DELEGATE}
unbuffer ${PROJ_ROOT}/scripts/build_ios.sh 2>&1 | ts
#store the binary
export PATH="$workdir/miniconda/bin:$PATH"
if [[ "$CIRCLE_BRANCH" == "nightly" ]]; then
export USE_PYTORCH_METAL_EXPORT=1
+ export USE_COREML_DELEGATE=1
fi
if [[ "$PACKAGE_TYPE" == conda ]]; then
"$workdir/builder/conda/build_pytorch.sh"
lite_interpreter:
type: string
default: "1"
+ use_coreml:
+ type: string
+ default: "0"
environment:
BUILD_ENVIRONMENT: << parameters.build_environment >>
IOS_ARCH: << parameters.ios_arch >>
SELECTED_OP_LIST: << parameters.op_list >>
USE_PYTORCH_METAL: << parameters.use_metal >>
BUILD_LITE_INTERPRETER: << parameters.lite_interpreter >>
+ USE_COREML_DELEGATE: << parameters.use_coreml >>
pytorch_windows_params: &pytorch_windows_params
parameters: