[audio] aarch64 support for audio extension

deepjavalibrary · Dec 20, 2022 · abb80f0 · abb80f0
1 parent ace20e1
commit abb80f0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/extensions/audio/build.gradle b/extensions/audio/build.gradle
@@ -3,7 +3,7 @@ group 'ai.djl.audio'
 dependencies {
     api project(":api")
     api project(":basicdataset")
-    api ("org.bytedeco:javacv:1.5.7") {
+    api ("org.bytedeco:javacv:1.5.8") {
         exclude group: "org.bytedeco", module: "opencv"
         exclude group: "org.bytedeco", module: "openblas"
         exclude group: "org.bytedeco", module: "flycapture"
@@ -24,6 +24,8 @@ dependencies {
     runtimeOnly "org.bytedeco:ffmpeg:5.0-1.5.7:macosx-x86_64"
     runtimeOnly "org.bytedeco:ffmpeg:5.0-1.5.7:linux-x86_64"
     runtimeOnly "org.bytedeco:ffmpeg:5.0-1.5.7:windows-x86_64"
+    runtimeOnly "org.bytedeco:ffmpeg:5.0-1.5.7:macosx-arm"
+    runtimeOnly "org.bytedeco:ffmpeg:5.0-1.5.7:linux-arm"
 
     testImplementation project(":testing")
     testRuntimeOnly "org.apache.logging.log4j:log4j-slf4j-impl:${log4j_slf4j_version}"

diff --git a/extensions/audio/src/test/java/ai/djl/audio/FFmpegAudioFactoryTest.java b/extensions/audio/src/test/java/ai/djl/audio/FFmpegAudioFactoryTest.java
@@ -17,7 +17,6 @@
 import ai.djl.ndarray.NDArray;
 import ai.djl.ndarray.NDManager;
 import ai.djl.ndarray.types.Shape;
-import ai.djl.testing.TestRequirements;
 import ai.djl.training.util.DownloadUtils;
 
 import org.testng.Assert;
@@ -33,7 +32,6 @@ public class FFmpegAudioFactoryTest {
 
     @BeforeClass
     public void setUp() throws IOException {
-        TestRequirements.notArm();
         DownloadUtils.download(URL, "build/test/test_01.wav");
     }
 

diff --git a/extensions/audio/src/test/java/ai/djl/audio/processor/AudioProcessorTest.java b/extensions/audio/src/test/java/ai/djl/audio/processor/AudioProcessorTest.java
@@ -19,7 +19,6 @@
 import ai.djl.ndarray.NDArray;
 import ai.djl.ndarray.NDManager;
 import ai.djl.ndarray.types.Shape;
-import ai.djl.testing.TestRequirements;
 import ai.djl.training.util.DownloadUtils;
 
 import org.testng.Assert;
@@ -36,7 +35,6 @@ public class AudioProcessorTest {
 
     @BeforeClass
     public void setUp() throws IOException {
-        TestRequirements.notArm();
         DownloadUtils.download(URL, "build/test/test_01.wav");
     }