Skip to content

Commit

Permalink
[pytorch] Use precxx11 build for aarch64 native library (#1619)
Browse files Browse the repository at this point in the history
Change-Id: I39f6da3444eb7ba5205bfde11f6c36e9baa76eab
  • Loading branch information
frankfliu authored May 3, 2022
1 parent 867dcd5 commit 06baa57
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion engines/pytorch/pytorch-jni/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ processResources {
"win-x86_64/cpu/djl_torch.dll"
]
if (ptVersion.startsWith("1.11.")) {
files.add("linux-aarch64/cpu/libdjl_torch.so")
files.add("linux-aarch64/cpu-precxx11/libdjl_torch.so")
files.add("linux-x86_64/cu113/libdjl_torch.so")
files.add("linux-x86_64/cu113-precxx11/libdjl_torch.so")
files.add("win-x86_64/cu113/djl_torch.dll")
Expand Down
6 changes: 3 additions & 3 deletions engines/pytorch/pytorch-native/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def prepareNativeLib(String binaryRoot, String ver) {
]

def aarch64Files = [
"${ver}/libtorch-cxx11-shared-with-deps-${ver}-aarch64.zip": "cpu/linux-aarch64"
"${ver}/libtorch-shared-with-deps-${ver}-aarch64.zip": "cpu-precxx11/linux-aarch64"
]

copyNativeLibToOutputDir(files, binaryRoot, officialPytorchUrl)
Expand Down Expand Up @@ -250,12 +250,12 @@ task uploadS3 {
"${BINARY_ROOT}/cpu/linux-x86_64/native/lib/",
"${BINARY_ROOT}/cpu/osx-x86_64/native/lib/",
"${BINARY_ROOT}/cpu/win-x86_64/native/lib/",
"${BINARY_ROOT}/cpu-precxx11/linux-aarch64/native/lib/",
"${BINARY_ROOT}/cpu-precxx11/linux-x86_64/native/lib/",
"${BINARY_ROOT}/cu102/linux/native/lib/",
"${BINARY_ROOT}/cu113/linux-x86_64/native/lib/",
"${BINARY_ROOT}/cu113/win-x86_64/native/lib/",
"${BINARY_ROOT}/cu113-precxx11/linux-x86_64/native/lib/",
"${BINARY_ROOT}/cpu/linux-aarch64/native/lib/"
"${BINARY_ROOT}/cu113-precxx11/linux-x86_64/native/lib/"
]
uploadDirs.each { item ->
fileTree(item).files.name.each {
Expand Down

0 comments on commit 06baa57

Please sign in to comment.