Add support for cuda version less then 10.0 (#3431)

ceb6265f · Pengfei Shi · GitHub · 12e220b7 · ceb6265f
Unverified Commit ceb6265f authored Oct 26, 2020 by Pengfei Shi Committed by GitHub Oct 26, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 3 deletions

include/LightGBM/cuda/vector_cudahost.h include/LightGBM/cuda/vector_cudahost.h +9 -3

No files found.
--- a/include/LightGBM/cuda/vector_cudahost.h
+++ b/include/LightGBM/cuda/vector_cudahost.h
@@ -65,9 +65,15 @@ struct CHAllocator {
      if (LGBM_config_::current_device == lgbm_device_cuda) {
        cudaPointerAttributes attributes;
        cudaPointerGetAttributes(&attributes, p);
-        if ((attributes.type == cudaMemoryTypeHost) && (attributes.devicePointer != NULL)) {
-          cudaFreeHost(p);
-        }
+        #if CUDA_VERSION >= 10000
+          if ((attributes.type == cudaMemoryTypeHost) && (attributes.devicePointer != NULL)) {
+            cudaFreeHost(p);
+          }
+        #else
+          if ((attributes.memoryType == cudaMemoryTypeHost) && (attributes.devicePointer != NULL)) {
+            cudaFreeHost(p);
+          }
+        #endif
      } else {
        _mm_free(p);
      }