def nth_chunk_for_pair(src, dst, idx):
    return (((src + (dst * collective.num_nodes)) * chunks) + idx)
def nth_chunk_for_pair(src, dst, idx):
    return (((src + (dst * collective.num_nodes)) * chunks) + idx)