From 8831bce10e6a45c2985e4b5c45960b03222c4870 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jakub=20Klinkovsk=C3=BD?= <klinkjak@fjfi.cvut.cz> Date: Fri, 21 Dec 2018 11:06:59 +0100 Subject: [PATCH] Fixed Partitioner::getOwner --- src/TNL/DistributedContainers/Partitioner.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TNL/DistributedContainers/Partitioner.h b/src/TNL/DistributedContainers/Partitioner.h index 3c8cace6fb..c42646f286 100644 --- a/src/TNL/DistributedContainers/Partitioner.h +++ b/src/TNL/DistributedContainers/Partitioner.h @@ -43,7 +43,7 @@ public: __cuda_callable__ static int getOwner( Index i, Index globalSize, int partitions ) { - return i * partitions / globalSize; + return min( partitions - 1, i / (globalSize / partitions) ); } // Gets the offset of data for given rank. -- GitLab