Merge pull request #6 from koide3/mem

cuda_mem_get_info
koide3 · Jul 20, 2024 · 71a104b · 71a104b
2 parents 17f78e8 + ac0d902
commit 71a104b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/gtsam_points/cuda/cuda_memory.hpp b/include/gtsam_points/cuda/cuda_memory.hpp
@@ -21,6 +21,8 @@ void cuda_host_to_device(void* dst, const void* src, size_t size, CUstream_st* s
 
 void cuda_device_to_host(void* dst, const void* src, size_t size, CUstream_st* stream = nullptr);
 
+void cuda_mem_get_info(size_t* free, size_t* total);
+
 template <typename T>
 T* cuda_new(CUstream_st* stream = nullptr) {
   return cuda_malloc(sizeof(T), stream);

diff --git a/src/gtsam_points/cuda/cuda_memory.cu b/src/gtsam_points/cuda/cuda_memory.cu
@@ -32,4 +32,8 @@ void cuda_device_to_host(void* dst, const void* src, size_t size, CUstream_st* s
   check_error << cudaMemcpyAsync(dst, src, size, cudaMemcpyDeviceToHost, stream);
 }
 
+void cuda_mem_get_info(size_t* free, size_t* total) {
+  check_error << cudaMemGetInfo(free, total);
+}
+
 }  // namespace gtsam_points