Ensure CUDA vector length is consistent with AlignedSize (#3748)

5784ffe7 · Chip Kerchner · GitHub · 1abc2e06 · 5784ffe7
Unverified Commit 5784ffe7 authored Jan 11, 2021 by Chip Kerchner Committed by GitHub Jan 11, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

include/LightGBM/cuda/vector_cudahost.h include/LightGBM/cuda/vector_cudahost.h +1 -0

No files found.
--- a/include/LightGBM/cuda/vector_cudahost.h
+++ b/include/LightGBM/cuda/vector_cudahost.h
@@ -42,6 +42,7 @@ struct CHAllocator {
  T* allocate(std::size_t n) {
    T* ptr;
    if (n == 0) return NULL;
+    n = (n + kAlignedSize - 1) & -kAlignedSize;
    #ifdef USE_CUDA
      if (LGBM_config_::current_device == lgbm_device_cuda) {
        cudaError_t ret = cudaHostAlloc(&ptr, n*sizeof(T), cudaHostAllocPortable);