From 37c5a6447fd6ab4cc4a7ad9541f71285a83ec87d Mon Sep 17 00:00:00 2001 From: Robert Maynard Date: Fri, 11 Oct 2024 10:37:44 -0400 Subject: [PATCH] Properly define block size for > sm_100 --- cpp/include/cuml/experimental/fil/detail/gpu_introspection.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/include/cuml/experimental/fil/detail/gpu_introspection.hpp b/cpp/include/cuml/experimental/fil/detail/gpu_introspection.hpp index b4d132de66..7261eb56f6 100644 --- a/cpp/include/cuml/experimental/fil/detail/gpu_introspection.hpp +++ b/cpp/include/cuml/experimental/fil/detail/gpu_introspection.hpp @@ -104,7 +104,7 @@ auto constexpr static const WARP_SIZE = index_type{32}; auto constexpr static const MAX_THREADS_PER_BLOCK = index_type{256}; #ifdef __CUDACC__ #if __CUDA_ARCH__ == 720 || __CUDA_ARCH__ == 750 || __CUDA_ARCH__ == 860 || \ - __CUDA_ARCH__ == 870 || __CUDA_ARCH__ == 890 + __CUDA_ARCH__ == 870 || __CUDA_ARCH__ == 890 || __CUDA_ARCH__ == 1200 auto constexpr static const MAX_THREADS_PER_SM = index_type{1024}; #else auto constexpr static const MAX_THREADS_PER_SM = index_type{2048};