github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(dfr): make HPX runtime environment persistent and deallocate temporaries.

Browse Source
This commit is contained in:
Antoniu Pop
2021-12-20 11:06:17 +00:00
committed by Antoniu Pop
parent 23d3dca3f2
commit bc3d647453
2 changed files with 48 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
compiler/include/zamalang/Runtime/distributed_generic_task_server.hpp
View File

@@ -25,6 +25,7 @@
#include "zamalang/Runtime/key_manager.hpp"
extern WorkFunctionRegistry *node_level_work_function_registry;
extern std::list<void *> new_allocated;
using namespace hpx::naming;
using namespace hpx::components;
@@ -95,6 +96,7 @@ struct OpaqueOutputData {
for (size_t i = 0; i < p; ++i)
ar &output[i];
outputs.push_back((void *)output);
new_allocated.push_back((void *)output);
}
alloc_p = true;
}
@@ -148,6 +150,7 @@ struct GenericComputeServer : component_base<GenericComputeServer> {
"Error: number of task parameters not supported.");
}
outputs = {output};
new_allocated.push_back(output);
break;
}
case 2: {
@@ -173,6 +176,8 @@ struct GenericComputeServer : component_base<GenericComputeServer> {
"Error: number of task parameters not supported.");
}
outputs = {output1, output2};
new_allocated.push_back(output1);
new_allocated.push_back(output2);
break;
}
case 3: {
@@ -199,6 +204,9 @@ struct GenericComputeServer : component_base<GenericComputeServer> {
"Error: number of task parameters not supported.");
}
outputs = {output1, output2, output3};
new_allocated.push_back(output1);
new_allocated.push_back(output2);
new_allocated.push_back(output3);
break;
}
default: