Method distribute_numpy

include/core/solver.h:1110–1138

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

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

        for (size_t i = start; i < end; i++) {
            Index head_global_id = array(i, 0);
            Index tail_global_id = array(i, 1);
            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;
            Index *_sample = reinterpret_cast<Index *>(&sample);
            for (int j = 0; j < kSampleSize; j++)
                _sample[j] = array(i, kSampleSize - j - 1);
            std::get<0>(sample) = head_local_id;
            std::get<1>(sample) = tail_local_id;

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

    /** @return GPU memory cost */
    static size_t gpu_memory_demand() {