| 204 | } |
| 205 | |
| 206 | void dealloc(mgbComputeNode_t device, void* addr) { |
| 207 | auto cn = *reinterpret_cast<mgb::CompNode*>(device); |
| 208 | XLAMemAllocHelper::inst()->dealloc(cn, addr); |
| 209 | } |
| 210 | |
| 211 | void log_xla_mem_states() { |
| 212 | XLAMemAllocHelper::inst()->log_states(); |