[pytorch] Fixes uploadS3 gradle task

deepjavalibrary · Jun 18, 2024 · 04d6561 · 04d6561
1 parent bb7dfd1
commit 04d6561
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/engines/pytorch/pytorch-engine/src/test/java/ai/djl/pytorch/integration/ALibUtilsTest.java b/engines/pytorch/pytorch-engine/src/test/java/ai/djl/pytorch/integration/ALibUtilsTest.java
@@ -27,7 +27,7 @@ public void setup() {
         TestRequirements.notMacX86();
 
         System.setProperty("ai.djl.pytorch.native_helper", ALibUtilsTest.class.getName());
-        System.setProperty("STDCXX_LIBRARY_PATH", "/usr/lib/non-exists");
+        System.setProperty("LIBSTDCXX_LIBRARY_PATH", "/usr/lib/non-exists");
     }
 
     @AfterClass

diff --git a/engines/pytorch/pytorch-native/build.gradle.kts b/engines/pytorch/pytorch-native/build.gradle.kts
@@ -254,16 +254,15 @@ tasks {
                 )
                 for (item in uploadDirs)
                     fileTree(item).files.map { it.name }.forEach {
-                        val out = item.relativeTo(File("${BINARY_ROOT}/")).absolutePath
-                        appendLine(out + URLEncoder.encode(it, "UTF-8"))
+                        val out = item.toRelativeString(File("${BINARY_ROOT}/"))
+                        appendLine(out + "/" + URLEncoder.encode(it, "UTF-8"))
                     }
             }
             exec {
                 commandLine("aws", "s3", "sync", "$BINARY_ROOT", "s3://djl-ai/publish/pytorch/$VERSION/")
             }
         }
     }
-
 }
 
 java {

diff --git a/engines/tensorflow/tensorflow-native/build.gradle.kts b/engines/tensorflow/tensorflow-native/build.gradle.kts
@@ -72,8 +72,8 @@ tasks {
                 )
                 for (item in uploadDirs)
                     fileTree(item).files.map { it.name }.forEach {
-                        val out = item.relativeTo(buildDirectory / "native/").absolutePath
-                        appendLine(out + it)
+                        val out = item.toRelativeString(buildDirectory / "native/")
+                        appendLine("$out/$it")
                     }
             }
             delete(