Method distribute

include/core/solver.h:1083–1107

Distribute edges to the sample pool. This function can be parallelized. */

    /** Distribute edges to the sample pool. This function can be parallelized. */
    void distribute(size_t start, size_t end, int id) {
        auto &offsets = solver->pool_offsets[id];

        for (size_t i = start; i < end; i++) {
            Index head_global_id = std::get<0>((*solver->samples)[i]);
            Index tail_global_id = std::get<1>((*solver->samples)[i]);
            std::pair<int, Index> head = solver->head_locations[head_global_id];
            std::pair<int, Index> tail = solver->tail_locations[tail_global_id];
            int head_partition_id = head.first;
            int tail_partition_id = tail.first;
            Index head_local_id = head.second;
            Index tail_local_id = tail.second;

            auto &pool = solver->predict_pool[head_partition_id][tail_partition_id];
            auto &indexes = solver->sample_indexes[head_partition_id][tail_partition_id];
            size_t &offset = offsets[head_partition_id][tail_partition_id];
            EdgeSample sample = (*solver->samples)[i];
            std::get<0>(sample) = head_local_id;
            std::get<1>(sample) = tail_local_id;

            pool[offset] = sample;
            indexes[offset] = i;
            offset++;
        }
    }

    /** Distribute edges to the sample pool. This function can be parallelized. */
    void distribute_numpy(size_t start, size_t end, int id) {