Docs.

shwina · Jun 26, 2019 · 17dbfa8 · 17dbfa8
1 parent e279ba3
commit 17dbfa8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/include/rmm/mr/default_memory_resource.hpp b/include/rmm/mr/default_memory_resource.hpp
@@ -24,8 +24,6 @@ namespace mr {
  * is not supplied. The initial default memory resource is a
  * `cuda_memory_resource`.
  *
- * The default memory resource is used for all temporary memory allocation.
- *
  * This function is thread-safe.
  *
  * @return device_memory_resource* Pointer to the current default memory

diff --git a/src/mr/default_memory_resource.cpp b/src/mr/default_memory_resource.cpp
@@ -28,6 +28,7 @@ device_memory_resource* initial_resource() {
   return &resource;
 }
 
+// Use an atomic to guarantee thread safety
 std::atomic<device_memory_resource*>& get_default() {
   static std::atomic<device_memory_resource*> res{initial_resource()};
   return res;