Merge pull request #20 from titsuki/use-1.7.6

Use xgboost v1.7.6
titsuki · Aug 26, 2023 · ee629c5 · ee629c5
2 parents 05e57c1 + 8e9514d
commit ee629c5
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 2 deletions.
diff --git a/lib/Algorithm/XGBoost/CustomBuilder.rakumod b/lib/Algorithm/XGBoost/CustomBuilder.rakumod
@@ -35,7 +35,7 @@ class Algorithm::XGBoost::CustomBuilder:ver<0.0.5>:auth<cpan:TITSUKI> is Distrib
             { module => "Zef::Service::Shell::curl" },
         ];
         my $fetcher      = Zef::Fetch.new(:backends(@fetch-backends));
-        my $uri          = 'https://github.com/dmlc/xgboost/releases/download/v1.4.2/xgboost.tar.gz';
+        my $uri          = 'https://github.com/dmlc/xgboost/releases/download/v1.7.6/xgboost.tar.gz';
         my $archive-file = "xgboost.tar.gz".IO.e
         ?? "xgboost.tar.gz"
         !! $fetcher.fetch(Candidate.new(:$uri), "xgboost.tar.gz");
@@ -47,7 +47,12 @@ class Algorithm::XGBoost::CustomBuilder:ver<0.0.5>:auth<cpan:TITSUKI> is Distrib
         my $archive-file-with-cwd = $*CWD.add($archive-file);
         my $extract-dir = $extractor.extract(Candidate.new(:uri($archive-file-with-cwd)), $*CWD);
         chdir("xgboost");
-        when self!is-osx { shell("brew install libomp && cmake . && make") }
+        when self!is-osx {
+            my $fh = open :w, "./cmake/xgboost-config.cmake.in";
+            $fh.say: "../../misc/xgboost-config.cmake.in".IO.slurp;
+            $fh.close;
+            shell("brew install libomp && cmake . && make")
+        }
         when self!is-linux { shell("cmake . && make") }
     }
     method !is-osx(--> Bool) { shell("uname", :out).out.slurp.trim.lc eq "darwin" }

diff --git a/misc/xgboost-config.cmake.in b/misc/xgboost-config.cmake.in
@@ -0,0 +1,28 @@
+@PACKAGE_INIT@
+
+set(USE_OPENMP @USE_OPENMP@)
+set(USE_CUDA @USE_CUDA@)
+set(USE_NCCL @USE_NCCL@)
+set(XGBOOST_BUILD_STATIC_LIB @BUILD_STATIC_LIB@)
+
+include(CMakeFindDependencyMacro)
+
+if (XGBOOST_BUILD_STATIC_LIB)
+  find_dependency(Threads)
+  if(USE_OPENMP)
+    find_dependency(OpenMP)
+  endif()
+  if(OpenMP_CXX_FOUND)
+    target_link_libraries(MyTarget PUBLIC OpenMP::OpenMP_CXX)
+  endif()
+  if(USE_CUDA)
+    find_dependency(CUDA)
+  endif()
+  # nccl should be linked statically if xgboost is built as static library.
+endif (XGBOOST_BUILD_STATIC_LIB)
+
+if(NOT TARGET xgboost::xgboost)
+  include(${CMAKE_CURRENT_LIST_DIR}/XGBoostTargets.cmake)
+endif()
+
+message(STATUS "Found XGBoost (found version \"${xgboost_VERSION}\")")