| 25 | } |
| 26 | |
| 27 | void sync(mgbComputeNode_t cn) { |
| 28 | if (!is_finalize()) { |
| 29 | auto* s = reinterpret_cast<mgb::CompNode*>(cn); |
| 30 | if (s->valid()) |
| 31 | s->sync(); |
| 32 | } |
| 33 | } |
| 34 | |
| 35 | mgbComputeNode_t load_cuda_cn(int device_id, int stream) { |
| 36 | std::string loc = ssprintf("gpu%i:%i", device_id, stream); |