diff options
Diffstat (limited to 'src/core/helpers/MemoryHelpers.h')
-rw-r--r-- | src/core/helpers/MemoryHelpers.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/core/helpers/MemoryHelpers.h b/src/core/helpers/MemoryHelpers.h index 6756a90c25..e751e6025d 100644 --- a/src/core/helpers/MemoryHelpers.h +++ b/src/core/helpers/MemoryHelpers.h @@ -46,6 +46,15 @@ using WorkspaceData = std::vector<std::pair<int, std::unique_ptr<TensorType>>>; template <typename TensorType> WorkspaceData<TensorType> manage_workspace(const experimental::MemoryRequirements &mem_reqs, MemoryGroup &mgroup, + ITensorPack &run_pack) +{ + ITensorPack dummy_pack = ITensorPack(); + return manage_workspace<TensorType>(mem_reqs, mgroup, run_pack, dummy_pack); +} + +template <typename TensorType> +WorkspaceData<TensorType> manage_workspace(const experimental::MemoryRequirements &mem_reqs, + MemoryGroup &mgroup, ITensorPack &run_pack, ITensorPack &prep_pack) { WorkspaceData<TensorType> workspace_memory; |