aboutsummaryrefslogtreecommitdiff
path: root/src/core/helpers/MemoryHelpers.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/helpers/MemoryHelpers.h')
-rw-r--r--src/core/helpers/MemoryHelpers.h9
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;