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