feat: add openmp feature

Author: thewh1teagle

llama-cpp-2/Cargo.toml

@@ -18,13 +18,16 @@ tracing = { workspace = true }
 encoding_rs = { workspace = true }
 
 [features]
+default = ["openmp"]
 cuda = ["llama-cpp-sys-2/cuda"]
 metal = ["llama-cpp-sys-2/metal"]
 dynamic-link = ["llama-cpp-sys-2/dynamic-link"]
 vulkan = ["llama-cpp-sys-2/vulkan"]
 native = ["llama-cpp-sys-2/native"]
+openmp = ["llama-cpp-sys-2/openmp"]
 sampler = []
 
+
 [target.'cfg(all(target_os = "macos", any(target_arch = "aarch64", target_arch = "arm64")))'.dependencies]
 llama-cpp-sys-2 = { path = "../llama-cpp-sys-2", version = "0.1.69", features = [
     "metal",

llama-cpp-sys-2/Cargo.toml

@@ -69,3 +69,4 @@ metal = []
 dynamic-link = []
 vulkan = []
 native = []
+openmp = []

llama-cpp-sys-2/build.rs

@@ -245,6 +245,10 @@ fn main() {
         config.define("GGML_CUDA", "ON");
     }
 
+    if cfg!(feature = "openmp") {
+        config.define("GGML_OPENMP", "ON");
+    }
+
     // General
     config
         .profile(&profile)
@@ -272,6 +276,13 @@ fn main() {
         );
     }
 
+    // OpenMP
+    if cfg!(feature = "openmp") {
+        if target.contains("gnu") {
+            println!("cargo:rustc-link-lib=gomp");
+        }
+    }
+
     // Windows debug
     if cfg!(all(debug_assertions, windows)) {
         println!("cargo:rustc-link-lib=dylib=msvcrtd");