Skip to content

Commit 55ce0bc

Browse files
authored
Merge pull request #396 from renato-arantes/pytorch_fp16
Enable fp16 linear layers in PyTorch via ACL
2 parents d19794c + 340d343 commit 55ce0bc

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ where `YY` is the year, and `MM` the month of the increment.
88
## [unreleased]
99

1010
### Added
11+
- Adds PyTorch [PR][https://github.com/pytorch/pytorch/pull/144992] Enable fp16 linear layers in PyTorch via ACL
1112

1213
### Changed
1314

ML-Frameworks/pytorch-aarch64/get-source.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,10 @@ git-shallow-clone https://github.com/pytorch/pytorch.git $PYTORCH_HASH
7676
# make default for AArch64 builds
7777
apply-github-patch pytorch/pytorch 9f6a4018f6e7d77d4ac974a38f68fbd7c8eef25c
7878

79+
# https://github.com/pytorch/pytorch/pull/144992 - Enable fp16 linear layers in PyTorch via ACL
80+
apply-github-patch pytorch/pytorch 00076d21ed6cd7df2a61165b1fb1d0a436f4e403
81+
apply-github-patch pytorch/pytorch 850db41fe6d33c6460740da781b40e009f04a47c
82+
7983
# Remove deps that we don't need for manylinux AArch64 CPU builds before fetching.
8084
# Only used when jni.h is present (see .ci/pytorch/build.sh:116), which is not the case for manylinux
8185
git rm android/libs/fbjni

0 commit comments

Comments
 (0)