Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-09-19 10:35:42 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

First implementation of hierarchical augmenter communication phase

Browse Source
This commit is contained in:
Gunther H. Weber 2021-05-17 21:05:14 -07:00
parent 00e906313f
commit e9af397fb2
1 changed files with 40 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
vtkm/filter/ContourTreeUniformDistributed.hxx
View File

@ -954,19 +954,53 @@ VTKM_CONT void ContourTreeUniformDistributed::DoPostExecute(
if (this->AugmentHierarchicalTree)
{
master.foreach (
[&](vtkm::worklet::contourtree_distributed::DistributedContourTreeBlockData<FieldType>*
blockData,
const vtkmdiy::Master::ProxyWithLink&) {
[](vtkm::worklet::contourtree_distributed::DistributedContourTreeBlockData<FieldType>*
blockData,
const vtkmdiy::Master::ProxyWithLink&) {
blockData->HierarchicalAugmenter.Initialize(
blockData->BlockIndex, &blockData->HierarchicalTree, &blockData->AugmentedTree);
});
// TODO/FIXME: Exchange
vtkmdiy::reduce(
master,
assigner,
partners,
[](vtkm::worklet::contourtree_distributed::DistributedContourTreeBlockData<FieldType>*
blockData,
const vtkmdiy::ReduceProxy& rp,
const vtkmdiy::RegularMergePartners&) {
auto round = rp.round();
const auto selfid = rp.gid();
for (int i = 0; i < rp.in_link().size(); ++i)
{
int ingid = rp.in_link().target(i).gid;
if (ingid == selfid)
{
worklet::contourtree_distributed::HierarchicalAugmenter<FieldType> inAugmenter;
rp.dequeue(ingid, inAugmenter);
blockData->HierarchicalAugmenter.RetrieveInAttachmentPoints(inAugmenter);
}
}
for (int i = 0; i < rp.out_link().size(); ++i)
{
auto target = rp.out_link().target(i);
if (target.gid != selfid)
{
blockData->HierarchicalAugmenter.PrepareOutAttachmentPoints(round);
// TODO/FIXME: Correct function? Correct round?
rp.enqueue(target, blockData->HierarchicalAugmenter);
blockData->HierarchicalAugmenter.ReleaseOutArrays(); // TODO/FIXME: Correct function?
}
}
});
master.foreach (
[&](vtkm::worklet::contourtree_distributed::DistributedContourTreeBlockData<FieldType>*
blockData,
const vtkmdiy::Master::ProxyWithLink&) {
[](vtkm::worklet::contourtree_distributed::DistributedContourTreeBlockData<FieldType>*
blockData,
const vtkmdiy::Master::ProxyWithLink&) {
blockData->HierarchicalAugmenter.BuildAugmentedTree();
});
}