| 47 | } |
| 48 | |
| 49 | void unload_cuda_cn(mgbComputeNode_t cn) { |
| 50 | auto* device = reinterpret_cast<mgb::CompNode*>(cn); |
| 51 | auto& cache = cn_cache(); |
| 52 | mgb_assert( |
| 53 | cache.find(device->to_string_physical()) != cache.end() && |
| 54 | device == cache[device->to_string_physical()]); |
| 55 | cache.erase(device->to_string_physical()); |
| 56 | delete device; |
| 57 | } |
| 58 | |
| 59 | void* get_cuda_stream(mgbComputeNode_t device) { |
| 60 | void* rst = nullptr; |
nothing calls this directly
no test coverage detected