diff --git a/engines/pytorch/pytorch-jni/build.gradle b/engines/pytorch/pytorch-jni/build.gradle index 1ea184810ec..e7441849025 100644 --- a/engines/pytorch/pytorch-jni/build.gradle +++ b/engines/pytorch/pytorch-jni/build.gradle @@ -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") diff --git a/engines/pytorch/pytorch-native/build.gradle b/engines/pytorch/pytorch-native/build.gradle index f1fdd975e55..77fb518402a 100644 --- a/engines/pytorch/pytorch-native/build.gradle +++ b/engines/pytorch/pytorch-native/build.gradle @@ -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) @@ -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 {