| 17 | } |
| 18 | |
| 19 | void WorkspaceWrapper::update(size_t size_in_bytes) { |
| 20 | mgb_assert(this->valid()); |
| 21 | if (size_in_bytes > m_workspace.size) { |
| 22 | // free workspace |
| 23 | if (m_workspace.size > 0) { |
| 24 | megdnn_free(m_handle, m_workspace.raw_ptr); |
| 25 | m_workspace.raw_ptr = nullptr; |
| 26 | } |
| 27 | // alloc new workspace |
| 28 | m_workspace.size = size_in_bytes; |
| 29 | if (m_workspace.size > 0) { |
| 30 | m_workspace.raw_ptr = |
| 31 | static_cast<dt_byte*>(megdnn_malloc(m_handle, size_in_bytes)); |
| 32 | } else { |
| 33 | m_workspace.raw_ptr = nullptr; |
| 34 | } |
| 35 | } |
| 36 | } |
| 37 | void WorkspaceWrapper::clear() { |
| 38 | if (m_workspace.size > 0) { |
| 39 | megdnn_free(m_handle, m_workspace.raw_ptr); |