| 6 | namespace test { |
| 7 | |
| 8 | class WorkspaceWrapper { |
| 9 | public: |
| 10 | WorkspaceWrapper(); |
| 11 | WorkspaceWrapper(Handle* handle, size_t size_in_bytes = 0); |
| 12 | ~WorkspaceWrapper(); |
| 13 | |
| 14 | WorkspaceWrapper& operator=(const WorkspaceWrapper& wrapper); |
| 15 | void clear(); |
| 16 | void update(size_t size_in_bytes); |
| 17 | |
| 18 | bool valid() const { return m_handle != nullptr; } |
| 19 | Workspace workspace() const { return m_workspace; } |
| 20 | |
| 21 | private: |
| 22 | Handle* m_handle; |
| 23 | Workspace m_workspace; |
| 24 | }; |
| 25 | |
| 26 | } // namespace test |
| 27 | } // namespace megdnn |