Merge pull request #688 from rapidsai/branch-22.06

[gpuCI] Forward-merge branch-22.06 to branch-22.08 [skip gpuci]
rapidsai · May 26, 2022 · f8c0676 · f8c0676
2 parents 9168ed8 + 120d478
commit f8c0676
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/cpp/include/raft/core/cudart_utils.hpp b/cpp/include/raft/core/cudart_utils.hpp
@@ -36,7 +36,6 @@
 #include <iomanip>
 #include <iostream>
 #include <mutex>
-#include <vector>
 
 ///@todo: enable once logging has been enabled in raft
 //#include "logger.hpp"
@@ -303,10 +302,10 @@ void print_device_vector(const char* variable_name,
                          size_t componentsCount,
                          OutStream& out)
 {
-  std::vector<T> host_mem(componentsCount);
-  CUDA_CHECK(
-    cudaMemcpy(host_mem.data(), devMem, componentsCount * sizeof(T), cudaMemcpyDeviceToHost));
-  print_host_vector(variable_name, host_mem.data(), componentsCount, out);
+  T* host_mem = new T[componentsCount];
+  CUDA_CHECK(cudaMemcpy(host_mem, devMem, componentsCount * sizeof(T), cudaMemcpyDeviceToHost));
+  print_host_vector(variable_name, host_mem, componentsCount, out);
+  delete[] host_mem;
 }
 
 /**