get a load-balanced and randomized distribution along one tensor dimension
Type | Intent | Optional | Attributes | Name | ||
---|---|---|---|---|---|---|
integer, | intent(in) | :: | nblk |
number of blocks (along one tensor dimension) |
||
integer, | intent(in) | :: | nproc |
number of processes (along one process grid dimension) |
||
integer, | intent(in), | DIMENSION(nblk) | :: | blk_size |
block sizes |
|
integer, | intent(out), | DIMENSION(nblk) | :: | dist |
distribution |