check for macos _or_ arm

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Mar 12, 2024 · 22fde50 · 22fde50
1 parent 1a0af3b
commit 22fde50
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle
@@ -51,8 +51,9 @@ dependencies {
     implementation("ai.djl.onnxruntime:onnxruntime-engine:0.21.0") {
         exclude group: "com.microsoft.onnxruntime", module: "onnxruntime"
     }
-    //arm doesn't support GPU
-    if (System.getProperty("os.arch") == "aarch64") {
+    def os = DefaultNativePlatform.currentOperatingSystem
+    //arm/macos doesn't support GPU
+    if (os.macOsX || System.getProperty("os.arch") == "aarch64") {
         dependencies {
             implementation "com.microsoft.onnxruntime:onnxruntime:1.14.0"
         }