Commit 726f444f authored by wooway777's avatar wooway777
Browse files

issue/21 - added zero workspace handling

parent d7965f91
...@@ -6,7 +6,7 @@ InferenceContext::InferenceContext(DeviceResource *rsrc, CacheManager *cache_man ...@@ -6,7 +6,7 @@ InferenceContext::InferenceContext(DeviceResource *rsrc, CacheManager *cache_man
: rsrc(rsrc), cache_manager(cache_manager), stream(stream) {} : rsrc(rsrc), cache_manager(cache_manager), stream(stream) {}
void InferenceContext::ensure_workspace(size_t required_size) { void InferenceContext::ensure_workspace(size_t required_size) {
if (required_size > current_workspace_size) { if (required_size > current_workspace_size || !workspace_storage) {
workspace_storage = Storage::createFromPool(required_size, rsrc->memory_pool); workspace_storage = Storage::createFromPool(required_size, rsrc->memory_pool);
current_workspace_size = required_size; current_workspace_size = required_size;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment