| 154 | // alloc_map. |
| 155 | |
| 156 | void rmm_wrap_finalize (void) |
| 157 | { |
| 158 | if (rmm_wrap_context != NULL) |
| 159 | { |
| 160 | RMM_WRAP_CHECK_CUDA(cudaStreamDestroy(rmm_wrap_context->main_stream)); |
| 161 | delete (rmm_wrap_context) ; |
| 162 | rmm_wrap_context = NULL ; |
| 163 | } |
| 164 | } |
| 165 | |
| 166 | //------------------------------------------------------------------------------ |
| 167 | // rmm_wrap_initialize: initialize the global rmm_wrap_context |