Tested successfully on MAC as well.

Author: cyccbxhl

xgboost-sys/build.rs

@@ -33,7 +33,8 @@ fn main() {
     #[cfg(not(feature = "cuda"))]
     let mut dst = Config::new(&xgb_root);
 
-    let dst = dst.uses_cxx11()
+    #[allow(unused_mut)]
+    let mut dst = dst.uses_cxx11()
         .define("BUILD_STATIC_LIB", "ON");
 
     #[cfg(target_os = "macos")]
@@ -54,16 +55,12 @@ fn main() {
 
     let bindings = bindgen::Builder::default()
         .header("wrapper.h")
+        .blocklist_item("std::__1.*")
         .clang_args(&["-x", "c++", "-std=c++11"])
         .clang_arg(format!("-I{}", xgb_root.join("include").display()))
         .clang_arg(format!("-I{}", xgb_root.join("rabit/include").display()))
         .clang_arg(format!("-I{}", xgb_root.join("dmlc-core/include").display()));
 
-    #[cfg(target_os = "linux")]
-    let bindings = bindings
-        .clang_arg(format!("-I/usr/include/c++/11"))
-        .clang_arg(format!("-I/usr/include/x86_64-linux-gnu/c++/11"));
-
     #[cfg(feature = "cuda")]
     let bindings = bindings.clang_arg("-I/usr/local/cuda/include");
     let bindings = bindings
@@ -83,6 +80,7 @@ fn main() {
     // link to appropriate C++ lib
     if target.contains("apple") {
         println!("cargo:rustc-link-lib=c++");
+        println!("cargo:rustc-link-search=native=/opt/homebrew/opt/libomp/lib");
         println!("cargo:rustc-link-lib=dylib=omp");
     } else {
         println!("cargo:rustc-cxxflags=-std=c++17");