lib/thundersvm.rb in thundersvm-0.1.4 vs lib/thundersvm.rb in thundersvm-0.2.0

@@ -17,10 +17,14 @@
   end
   lib_name =
     if Gem.win_platform?
       "thundersvm.dll"
     elsif RbConfig::CONFIG["host_os"] =~ /darwin/i
-      "libthundersvm.dylib"
+      if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
+        "libthundersvm.arm64.dylib"
+      else
+        "libthundersvm.dylib"
+      end
     else
       "libthundersvm.so"
     end
   vendor_lib = File.expand_path("../vendor/#{lib_name}", __dir__)
   self.ffi_lib = [vendor_lib]