Loading src/TNL/DistributedContainers/Partitioner.h +1 −1 Original line number Diff line number Diff line Loading @@ -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. Loading Loading
src/TNL/DistributedContainers/Partitioner.h +1 −1 Original line number Diff line number Diff line Loading @@ -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. Loading