add max shared mem workaround

2023-03-19 20:08:42 +01:00 · 2023-03-19 20:08:42 +01:00 · 595cd81251
commit 595cd81251
parent 0325ce0bed
1 changed files with 4 additions and 0 deletions
--- a/src/tree/gpu_hist/histogram.cu
+++ b/src/tree/gpu_hist/histogram.cu
@ -278,7 +278,11 @@ void BuildGradientHistogram(CUDAContext const* ctx, EllpackDeviceAccessor const&
 #endif

  // opt into maximum shared memory for the kernel if necessary
+#if defined(XGBOOST_USE_CUDA)
  size_t max_shared_memory = dh::MaxSharedMemoryOptin(device);
+#elif defined(XGBOOST_USE_HIP)
+  size_t max_shared_memory = dh::MaxSharedMemory(device);
+#endif

  size_t smem_size =
      sizeof(GradientPairInt64) * feature_groups.max_group_bins;