1. 08 Jun, 2019 1 commit
  2. 07 Jun, 2019 1 commit
  3. 21 May, 2019 1 commit
    • Minjie Wang's avatar
      [Refactor] Immutable graph index (#543) · 605b5185
      Minjie Wang authored
      * WIP
      
      * header
      
      * WIP .cc
      
      * WIP
      
      * transpose
      
      * wip
      
      * immutable graph .h and .cc
      
      * WIP: nodeflow.cc
      
      * compile
      
      * remove all tmp dl managed ctx; they caused refcount issue
      
      * one simple test
      
      * WIP: testing
      
      * test_graph
      
      * fix graph index
      
      * fix bug in sampler; pass pytorch utest
      
      * WIP on mxnet
      
      * fix lint
      
      * fix mxnet unittest w/ unfortunate workaround
      
      * fix msvc
      
      * fix lint
      
      * SliceRows and test_nodeflow
      
      * resolve reviews
      
      * resolve reviews
      
      * try fix win ci
      
      * try fix win ci
      
      * poke win ci again
      
      * poke
      
      * lazy multigraph flag; stackoverflow error
      
      * revert node subgraph test
      
      * lazy object
      
      * try fix win build
      
      * try fix win build
      
      * poke ci
      
      * fix build script
      
      * fix compile
      
      * add a todo
      
      * fix reviews
      
      * fix compile
      605b5185
  4. 08 Apr, 2019 1 commit
    • Da Zheng's avatar
      [Feature] Create shared memory graph store. (#468) · bfdd1eaa
      Da Zheng authored
      * accelerate gcn_ns.
      
      * add timing.
      
      * run infer with whole graph.
      
      * distributed gcn_ns.
      
      * reconstruct gcn_ns.
      
      * minor fix.
      
      * change graphsage_cv for numa.
      
      * fix #OMP threads.
      
      * accelerate graphsage_cv.
      
      * fix a weird bug.
      
      * add profiler in graphsage_cv.
      
      * accelerate graphsage_cv.
      
      manually aggregate neighbors' embeddings with pull.
      
      * load csr directly in gcn_ns_sc.
      
      * parallel sort for graph index.
      
      * Revert "parallel sort for graph index."
      
      This reverts commit 86fe2c7117fe5e56b0d481b39849c258b166945b.
      
      * run gcn_ns_sc on GPUs.
      
      * acc gcn_cv_sc.
      
      * change gcn_cv for numa.
      
      * fix gcn_cv to use numa and gpu.
      
      * improve graphsage_cv to use numa and gpu.
      
      * improve gcn_ns.
      
      * improve graphsage_cv.
      
      * init shared memory graph store.
      
      * fix.
      
      * enable init ndata.
      
      * improve tests.
      
      * add bidirectional communication.
      
      * link to rt.
      
      * fix compilation error.
      
      * fix shared memory init.
      
      * use MessageQueue for inter-process communication.
      
      * reconstruct immutable graph csr.
      
      * fix gcn.
      
      * load csr to shared memory.
      
      * fix minor bugs.
      
      * add comments.
      
      * refactor SharedMemory.
      
      * fix bugs in ImmutableGraph.
      
      * create CSR graph from shared memory.
      
      * add more test for loading a csr graph.
      
      * terminate graph store properly.
      
      * allow initializing ndata in the graph store server.
      
      * use RPC for inter-process communication.
      
      * a script for loading a graph.
      
      * allow customizing port.
      
      * list all ndata and edata.
      
      * support dtype.
      
      * reorganize SharedMemoryGraphStore.
      
      * fix ndata shape.
      
      * reconstruct gcn_ns.
      
      * print info.
      
      * set omp in gcn_ns.
      
      * reset sampling examples.
      
      * fix lint.
      
      * fix lint.
      
      * reset gcn.
      
      * disable shared memory in windows.
      
      * fix.
      
      * fix.
      
      * reset changes.
      
      * revert nodeflow changes.
      
      * fix cmake.
      
      * fix test.
      
      * fix test.
      
      * fix test.
      
      * fix test.
      
      * add comments.
      
      * fix test.
      
      * move vector out.
      
      * fix lint.
      
      * fix lint.
      
      * move SharedMemory.
      
      * update cmake.
      
      * update comment.
      
      * fix comments.
      
      * Revert "update cmake."
      
      This reverts commit 592445e37077f70a6e3f2e5245f9a3d086b04f3b.
      
      * update cmake.
      
      * add comments.
      
      * rename.
      
      * change the comment.
      
      * fix a bug.
      
      * rename.
      
      * add comments.
      
      * add comments.
      
      * add init_edata.
      
      * rewrite memory alloc.
      
      * move vector to CSR.
      
      * fix.
      
      * init data.
      
      * Revert "init data."
      
      This reverts commit 2b217b9553911b7dd84a9f1d9b68430b5aa18e23.
      
      * init data.
      
      * init new columns correctly.
      bfdd1eaa
  5. 29 Mar, 2019 2 commits
    • Chao Ma's avatar
      [RPC] add C++ RPC infrastructure and distributed sampler (#465) · 5d494c62
      Chao Ma authored
      * add C++ rpc infrastructure and distributed sampler
      
      * update
      
      * update lint
      
      * update lint
      
      * update lint
      
      * update
      
      * update
      
      * update
      
      * updare
      
      * update
      
      * update
      
      * update
      
      * update serialize and unittest
      
      * update serialize
      
      * lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update unittest
      
      * put Finalize() to __del__
      
      * update unittest
      
      * update
      
      * delete buffer in Finalize
      
      * update unittest
      
      * update unittest
      
      * update unittest
      
      * update unittest
      
      * update
      
      * update
      
      * fix small bug
      
      * windows socket impl
      
      * update API
      
      * fix bug in serialize
      
      * fix bug in serialzie
      
      * set parent graph
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * fix lint
      
      * fix
      
      * fix windows compilation error
      
      * fix windows error
      
      * change API to lower-case
      
      * update test
      
      * fix typo
      
      * update
      
      * add SamplerPool
      
      * add SamplerPool
      
      * update
      
      * update test
      
      * update
      
      * update
      
      * update
      
      * update
      
      * add example
      
      * update
      
      * update
      5d494c62
    • Quan (Andy) Gan's avatar
      [Model][Feature] PinSage & Random Walk with Restart (#453) · 6066fee9
      Quan (Andy) Gan authored
      * random walk traces generation
      
      * remove outdated comments
      
      * oops put in the wrong place
      
      * explicit inline
      
      * moving rand_r to util
      
      * pinsage-like model on movielens
      
      * the code runs now
      
      * support cuda
      
      * using readonly graph
      
      * moving random walk to public function
      
      * per-thread seed and openmp support
      
      * pinsage-like model on movielens
      
      * the code runs now
      
      * support cuda
      
      * using readonly graph
      
      * using C random walk
      
      * removing profile decorators
      
      * param initialization
      
      * no grad
      
      * leaky relu fixes everything
      
      * train and save
      
      * WIP
      
      * WIP
      
      * WIP
      
      * seems to work
      
      * evaluation output
      
      * swapping order of val/test and train
      
      * debug
      
      * hyperparam tuning
      
      * prior/training dataset split changes
      
      * random walk reorg
      
      * random walk with restart
      
      * signed comparison fix
      
      * migrating random walk to nodeflow
      
      * Revert "migrating random walk to nodeflow"
      
      This reverts commit f2565347cced7c912a58a529b257c033d9f375b7.
      
      * add README and remove dataset
      
      * new endpoint
      
      * lint
      
      * lint x2
      
      * oops forgot test
      
      * including bpr - better for baseline
      
      * addressing fixes
      
      * throwing random walks out from SamplerOp class
      
      * forgot to move RandomWalk; why did this even work?
      
      * removing legacy garbage
      
      * add todo
      
      * address comments
      
      * stupid bug fix
      
      * call ndarrayvector converter to handle traces
      6066fee9
  6. 05 Mar, 2019 2 commits
  7. 03 Mar, 2019 1 commit
    • GaiYu0's avatar
      [Feature]Uniform layer-wise sampler (#416) · fb4246e5
      GaiYu0 authored
      * migrate to node-flow
      
      * uniform layer sampler test cases
      
      * more test cases
      
      * documentations
      
      * fix lint errors
      
      * fix lint errors
      
      * fix lint errors
      
      * iota
      
      * add asnumpy
      
      * requested changes
      
      * fix indptr error
      
      * fix lint errors
      
      * requested changes & fix lint errors
      
      * fix lint errors
      
      * fix LayerSampler unit test
      fb4246e5
  8. 28 Feb, 2019 1 commit
    • Ziyue Huang's avatar
      [Model][MXNet] neighbor sampling & skip connection & control variate & graphsage (#322) · 7e30382e
      Ziyue Huang authored
      * neighbor sampling draft
      
      * val/test acc
      
      * control variate draft
      
      * control variate
      
      * update
      
      * fix new_history
      
      * maintain aggregated history while updating new history
      
      * preprocess the first layer, change push to pull
      
      * update
      
      * fix subg_degree
      
      * nodeflow
      
      * clear
      
      * readme
      
      * doc and unittest for self loop
      
      * address comments
      
      * rename
      
      * update
      
      * fix
      
      * Update node_flow.py
      
      * Update node_flow.py
      7e30382e
  9. 25 Feb, 2019 1 commit
    • Minjie Wang's avatar
      [WIP] [NN] Refactor NN package (#406) · 565f0c88
      Minjie Wang authored
      * refactor graph conv
      
      * docs & tests
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix lint script
      
      * fix lint
      
      * Update
      
      * Style fix
      
      * Fix style
      
      * Fix style
      
      * Fix gpu case
      
      * Fix for gpu case
      
      * Hotfix edgesoftmax docs
      
      * Handle repeated features
      
      * Add docstring
      
      * Set default arguments
      
      * Remove dropout from nn.conv
      
      * Fix
      
      * add util fn for renaming
      
      * revert gcn_spmv.py
      
      * mx folder
      
      * fix wierd bug
      
      * fix mx
      
      * fix lint
      565f0c88
  10. 20 Feb, 2019 3 commits
  11. 19 Feb, 2019 1 commit
    • Da Zheng's avatar
      [Feature] add NodeFlow API (#361) · f370e628
      Da Zheng authored
      * sample layer subgraphs.
      
      * fix.
      
      * fix.
      
      * add layered subgraph.
      
      * fix lint.
      
      * fix.
      
      * fix tutorial.
      
      * fix.
      
      * remove copy_to_parent.
      
      * add num_layers
      
      * move sampling code to sampler.cc
      
      * fix.
      
      * move subgraph construction out.
      
      * Revert "move subgraph construction out."
      
      This reverts commit 24b3d13b0d8ed5f91847ea75a7674ee8f7d45cff.
      
      * change to NodeFlow.
      
      * use NodeFlow in Python.
      
      * use NodeFlowIndex.
      
      * add node_mapping and edge_mapping.
      
      * remove unnecessary code in SSE tutorial.
      
      * Revert "remove unnecessary code in SSE tutorial."
      
      This reverts commit 093f0413d5fa2e63ca5f80c46c80a126a9fb720c.
      
      * fix tutorial.
      
      * move to node_flow.
      
      * update gcn cv updater.
      
      * import NodeFlow.
      
      * update.
      
      * add demo code for vanilla control variate sampler.
      
      * update.
      
      * update.
      
      * add neighbor sampling.
      
      * return flow offsets.
      
      * update node_flow.
      
      * add test.
      
      * fix sampler.
      
      * fix graph index.
      
      * fix a bug in sampler.
      
      * fix map_to_layer_nid and map_to_flow_eid.
      
      * fix apply_flow.
      
      * remove model code.
      
      * implement flow_compute.
      
      * fix a bug.
      
      * reverse the csr physically.
      
      * add mini-batch test.
      
      * add mini batch test.
      
      * update flow_compute.
      
      * add prop_flows
      
      * run on specific nodes.
      
      * test copy
      
      * fix a bug in creating frame in NodeFlow.
      
      * add init gcn_cv_updater.
      
      * fix a minor bug.
      
      * fix gcn_cv_updater.
      
      * fix a bug.
      
      * fix a bug in NodeFlow.
      
      * use new h in gcn_cv_updater.
      
      * add layer_in_degree and layer_out_degree.
      
      * fix gcn_cv_updater for gpu.
      
      * temp fix in NodeFlow for diff context.
      
      * allow enabling/disabling copy back.
      
      * add with-updater option.
      
      * fix a bug in computing degree.
      
      * add with-cv option.
      
      * rename and add comments.
      
      * fix lint complain.
      
      * fix lint.
      
      * avoid assert.
      
      * remove assert.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * fix the methods in NodeFlow.
      
      * fix lint.
      
      * update SSE.
      
      * remove gcn_cv_updater.
      
      * correct comments for the schedulers.
      
      * update comment.
      
      * add map_to_nodeflow_nid
      
      * address comment.
      
      * remove duplicated test.
      
      * fix int.
      
      * fix comments.
      
      * fix lint
      
      * fix.
      
      * replace subgraph with NodeFlow.
      
      * move view.
      
      * address comments.
      
      * fix lint.
      
      * fix lint.
      
      * remove static_cast.
      
      * fix docstring.
      
      * fix comments.
      
      * break SampleSubgraph.
      
      * move neighbor sampling to sampler.cc
      
      * fix comments.
      
      * rename.
      
      * split neighbor_list.
      
      * address comments.
      
      * fix.
      
      * remove TODO.
      f370e628