| 40 | } |
| 41 | |
| 42 | void StaticDeviceMemoryManager::exec_exit() { |
| 43 | mgb_assert(m_in_exec.test_and_set()); |
| 44 | m_in_exec.clear(); |
| 45 | } |
| 46 | |
| 47 | const DeviceTensorStorage& StaticDeviceMemoryManager::alloc( |
| 48 | ComputingGraph* graph, CompNode cn, size_t size, size_t cur_version) { |