Class WorkerMixin

include/core/solver.h:1171–1623

 */
template<class _Solver>
class WorkerMixin {
public:
    typedef _Solver Solver;
    typedef typename Solver::Float Float;
    typedef typename Solver::Index Index;
    typedef typename Solver::Vector Vector;
    typedef typename Solver::EdgeSample EdgeSample;

    static const int kSampleSize = Solver::kSampleSize;

    Solver *solver;
    int device_id;
    Optimizer optimizer;
    cudaStream_t work_stream, sample_stream;
    Index head_partition_size, tail_partition_size;
    std::vector<std::shared_ptr<Memory<Vector, Index>>> embeddings, gradients;
    std::vector<std::shared_ptr<std::vector<Memory<Vector, Index>>>> moments;
    int head_partition_id, tail_partition_id;
    std::vector<Index> head_global_ids, tail_global_ids;
    std::vector<Protocol> protocols;
    Protocol sampler_protocol;
    AliasTable<Float, Index> negative_sampler;
    Memory<Index, int> batch, negative_batch;
    Memory<Float, int> logits, loss;
    Memory<double, int> random;
    curandGenerator_t generator;
    int num_moment;
    int num_embedding, num_negative, batch_size;
    int log_frequency;

#define USING_WORKER_MIXIN(type) \
    using typename type::Solver; \
    using typename type::Float; \
    using typename type::Index; \
    using typename type::Vector; \
    using type::solver; \
    using type::work_stream; \
    using type::head_partition_size; \
    using type::tail_partition_size; \
    using type::head_partition_id; \
    using type::tail_partition_id; \
    using type::head_global_ids; \
    using type::tail_global_ids; \
    using type::negative_sampler; \
    using type::batch; \
    using type::negative_batch; \
    using type::logits; \
    using type::loss; \
    using type::batch_size; \
    using type::embeddings; \
    using type::moments; \
    using type::gradients; \
    using type::num_moment; \
    using type::optimizer

    /**
     * Construct a general training worker
     * @param _solver pointer to graph embedding solver