Fix onnx dep (#2198) (#2201)

* onnxruntime depends on architecture, not os Signed-off-by: HenryL27 <[email protected]> * check for macos _or_ arm Signed-off-by: HenryL27 <[email protected]> --------- Signed-off-by: HenryL27 <[email protected]> (cherry picked from commit 4af7e78) Co-authored-by: HenryL27 <[email protected]>
opensearch-project · Mar 13, 2024 · 23b7bb9 · 23b7bb9
1 parent aaa2911
commit 23b7bb9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle
@@ -52,8 +52,8 @@ dependencies {
         exclude group: "com.microsoft.onnxruntime", module: "onnxruntime"
     }
     def os = DefaultNativePlatform.currentOperatingSystem
-    //mac doesn't support GPU
-    if (os.macOsX) {
+    //arm/macos doesn't support GPU
+    if (os.macOsX || System.getProperty("os.arch") == "aarch64") {
         dependencies {
             implementation "com.microsoft.onnxruntime:onnxruntime:1.14.0"
         }
@@ -122,4 +122,4 @@ spotless {
 
         eclipse().configFile rootProject.file('.eclipseformat.xml')
     }
-}
+}