[ci] Fixes build on macOS aarch64 machine (#3191)

deepjavalibrary · May 14, 2024 · 84d3581 · 84d3581
1 parent e62674c
commit 84d3581
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java b/examples/src/test/java/ai/djl/examples/inference/TimeSeriesTest.java
@@ -55,7 +55,7 @@ public void testM5Forecasting() throws ModelException, TranslateException, IOExc
 
     @Test
     public void testAirPassenger() throws ModelException, TranslateException, IOException {
-        TestRequirements.engine("MXNet");
+        TestRequirements.linux();
 
         float[] result = AirPassengersDeepAR.predict();
         logger.info("{}", result);

diff --git a/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java b/examples/src/test/java/ai/djl/examples/training/TrainResNetTest.java
@@ -44,7 +44,7 @@ public void testTrainResNet() throws ModelException, IOException, TranslateExcep
     @Test
     public void testTrainResNetSymbolicNightly()
             throws ModelException, IOException, TranslateException {
-        TestRequirements.engine("MXNet");
+        TestRequirements.linux();
         TestRequirements.nightly();
         TestRequirements.gpu("MXNet");
 
@@ -65,6 +65,7 @@ public void testTrainResNetSymbolicNightly()
     @Test
     public void testTrainResNetImperativeNightly()
             throws ModelException, IOException, TranslateException {
+        TestRequirements.linux();
         TestRequirements.nightly();
         TestRequirements.gpu("MXNet");
 

diff --git a/integration/build.gradle b/integration/build.gradle
@@ -10,7 +10,10 @@ dependencies {
     implementation project(":model-zoo")
     implementation project(":testing")
 
-    runtimeOnly project(":engines:mxnet:mxnet-model-zoo")
+    // Don't use MXNet for aarch64
+    if (System.properties['os.arch'] != "aarch64") {
+        runtimeOnly project(":engines:mxnet:mxnet-model-zoo")
+    }
     runtimeOnly project(":engines:pytorch:pytorch-model-zoo")
     runtimeOnly project(":engines:pytorch:pytorch-jni")
     runtimeOnly project(":engines:tensorflow:tensorflow-model-zoo")