GitLab

* [Model] SchNet

* [Model] SchNet model

* [Model] Schnet Model fix device-related bug

* [Model] SchNet fix bugs

* [Model] SchNet fix some bugs

* [Model] Schnet 🎨 code indent format

* [Model] SchNet ✏ fix some typos

* fix NDArray reference

* fix lint

* capture NDArray in the closure

* nonuniform sampler

* unit test

* test on out neighbors

* error checks

* lint

* fix

* clarification

* use macro switcher

* use empty array for uniform sampling

* oops

* Revert "oops"

This reverts commit a11f9ae707aaeb67fb5921c887a17d3711d5b04a.

* Revert "use empty array for uniform sampling"

This reverts commit 8526ce4cade89f2c1b09a08aca8830375ebafb31.

* re-reverting

* use a method

* removal doc

* glob

* upd

* rm knn

* add softmax

* upd

* upd

* add broadcast and s2s

* optimize max_on

* forsaken changes to heterograph

* upd

* upd

* upd

* upd

* upd

* bugfix

* upd

* upd

* upd

* upd

* format upd

* upd format

* upd doc

* upd

* import order

* upd

* rm warnings

* fix

* upd test

* upd

* upd

* fix device

* upd

* upd

* upd

* upd

* remove 1.1

* upd

* trigger

* trigger

* add more tests

* fix device

* upd

* upd

* refactor

* fix?

* fix

* upd docstring

* refactor

* upd

* fix

* upd

* upd

* upd

* fix

* upd docs

* add shape

* refactor & upd doc

* upd doc

* upd

* Refactoring Communicator

* fix lint

* change non-const reference

* add header file

* use MemoryBuffer

* update PR

* fix bug on csr shape

* zero-copy msg_queue

* fix lint

* fix lint

* fix lint

* add header file

* fix windows build error

* fix windows build error

* update

* fix lint

* update

* fix lint

* fix lint

* add more test

* fix windows test

* update windows test

* update windows test

* update windows test

* update

* fix lint

* fix lint

* update

* update

* update

* update

* use STATUS code

* update test

* remove mem_cpy

* fix lint

* update

* finish

* ConstructNFTensor

* add test for deallocator

* update

* fix lint

* add local scope function

* fix lint

* fix docstring

* change local_scope to local_var; add context manager

* address comments

* rng refactor

* fix bugs

* unit test

* remove setsize

* lint

* fix test

* use explicit instantiation instead of inlining

* stricter test

* use tvm solution

* moved python interface to dgl.random

* lint

* address comments

* make getthreadid an inline function

* return 1 option in scipy adjacency matrix

* lint

* use dgl warning

* i'm an idiot

* lint x2

* rename

* finish bipartite graph implementation; compiled

* finished heterograph implementation; compiled

* WIP: apis

* C API codes

* compiled

* WIP: python

* HeteroGraphIndex

* WIP: test

* add DGLContext support in ffi

* fix bug in has edge

* unittests except edge subgraph

* edge subgraph

* fix lint

* address comments

* poke ci

* try fix

* fix msvc

* WIP: using object system for graph

* c++ side refactoring done; compiled

* remove stale apis

* fix bug in DGLGraphCreate; passed test_graph.py

* fix bug in python modify; passed utest for pytorch/cpu

* fix lint

* address comments

* WIP: array refactoring

* WIP: implementation

* wip

* most csr part

* WIP: on coo

* WIP: coo

* finish refactoring immutable graph

* compiled

* fix undefined ndarray copy bug; add COOToCSR when coo has no data array

* fix bug in COOToCSR

* fix bug in CSR constructor

* fix bug in in_edges(vid)

* fix OutEdges bug

* pass test_graph

* pass test_graph

* fix bug in CSR constructor

* fix bug in CSR constructor

* fix bug in CSR constructor

* fix stupid bug

* pass gpu test

* remove debug printout

* fix lint

* rm biparate grpah

* fix lint

* address comments

* fix bug in Clone

* cpp utests

* try venv for win64

* update

* update

* trying another

* update

* update

* update

* trigger ci

* update

* update

* formatting

* formatting

* WIP: import tvm runtime node system

* WIP: object system

* containers

* tested basic container composition

* tested custom object

* fix setattr bug

* tested object container return

* fix lint

* some comments about get/set state

* fix lint

* fix lint

* update cython

* fix cython

* ffi doc

* fix doc

* diffpool original file added

* make diffpool fuse up and running

* minor tweak on tu dataset statistics method

* fix tu

* break

* delete break

* pre_org

* diffpool fuse reorg

* fix random shuffling

* fix bn

* add dgl layers

* early stopping

* add readme

* fix

* add diffpool preprocess script

* tweak tu dataset

* tweak

* tweak

* tweak

* tweak

* tweak

* preprocess dataset

* fix early stopping

* fix

* fix

* fix

* tweak

* readme

* code review

* code review

* dataset code review

* update README

* code review

* tu doc

* Add serialization

* fix memory leak

* Revert "Add serialization"

This reverts commit e838bb1faa2503d14593d702be00514f936c9027.

* fix

* c++ heterograph interfaces

* addressing comments

* update hetero api

* missing interfaces

* fix lint

* upd

* Add stronger test

* Update test_nn.py

* Update test_nn.py

* trigger

* trigger

* trigger

* Add serialization

* fix broken badge

* Revert "Add serialization"

This reverts commit e838bb1faa2503d14593d702be00514f936c9027.

* fix

* [Feature][RFC] Heterogeneous graph interface (#553)

* heterogeneous graph interface

* lint

* disable lints

* disable lint checks

* change node_types to dict

* update

* update

* heterograph view

* message passing with types

* clarifications

* graph queries

* clarifications

* moving add_XXX to Base

* from_networkx

* register functions

* Update heterograph.py

* lint.

* update speed numbers in readme

* numbers in examples/pytorch/readme

* use new split.

* fix.

* fix.

* add zero copy from numpy.

* workspace and disable mx tutorial

* try cleanws

* poke

* poke

* poke

* enable tutorial test in CI.

* reduce overhead.

* bump up version

* conda+cuda trial

* switch conda branch

* revert

* disable cudnn

* windows cuda arch fix

* more fix

* give warning.

* fix.

* update error msg.

* bump up version

* conda+cuda trial

* switch conda branch

* revert

* disable cudnn

* copy graph index to shared memory.

* fix.

* fix.

* fix.

* use a diff name for in-csr and out-csr.

* fix lint.

* remove print.

* add test.

* add comments.