Several changes to make CI pass.

rapidsai · Apr 2, 2024 · bdae07d · bdae07d
1 parent 7fd5382
commit bdae07d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 13 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -19,7 +19,7 @@ cmake_minimum_required(VERSION 3.23.1 FATAL_ERROR)
 
 include(FetchContent)
 
-project(GPUTreeShap VERSION 23.08.00 LANGUAGES CXX CUDA)
+project(GPUTreeShap VERSION 23.08.00 LANGUAGES C CXX CUDA)
 set(CMAKE_CXX_STANDARD 17)
 set(CMAKE_CUDA_STANDARD 17)
 
@@ -66,7 +66,7 @@ if(BUILD_GTEST)
   if(NOT MSVC)
     target_compile_options(TestGPUTreeShap PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:${GCC_COMPILE_OPTIONS}>)
   endif()
-  target_link_libraries(TestGPUTreeShap  PRIVATE gtest gtest_main)
+  target_link_libraries(TestGPUTreeShap PRIVATE GTest::gtest GTest::gtest_main)
 endif()
 
 if(BUILD_EXAMPLES)

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -46,17 +46,9 @@ dependencies:
               - cuda-nvcc
               - cuda-cudart-dev
           - matrix:
-              arch: x86_64
-              cuda: "11.8"
-            packages:
-              - nvcc_linux-64=11.8
-              - cudatoolkit
-          - matrix:
-              arch: aarch64
-              cuda: "11.8"
+              cuda: "11.*"
+            # GPUTreeShap CI uses the container's CUDA 11.8.
             packages:
-              - nvcc_linux-aarch64=11.8
-              - cudatoolkit
   docs:
     common:
       - output_types: conda

diff --git a/tests/test_gpu_treeshap.cu b/tests/test_gpu_treeshap.cu
@@ -25,7 +25,7 @@
 #include <numeric>
 #include <random>
 #include <vector>
-#include "gtest/gtest.h"
+#include <gtest/gtest.h>
 #include "tests/test_utils.h"
 #include "../GPUTreeShap/gpu_treeshap.h"