GitLab

Co-authored-by: xliucr <xliucr@amazon.com>

* Add spmm bf16 support and add ut

* Add bf16 support for spspmm

* Disable bf16 test before torch_scatter 2.0.9

* Refactor version compare

* Update test/utils.py
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

Co-authored-by: Dong Wang <d@dongs-mbp.lan>

Co-authored-by: Dong Wang <d@dongs-mbp.lan>

* disable undirected for temporal sampling

* disjoint sampling for temporal

* fix repeated node index

* compile fix

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* comments on directed to be true

* add directed in API

* comments

* minor function signature fix

* Update csrc/cpu/neighbor_sample_cpu.cpp

* Update csrc/neighbor_sample.cpp

* minor refactor

* minor refactor
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* disable undirected for temporal sampling

* disjoint sampling for temporal

* fix repeated node index

* compile fix

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>

* comments on directed to be true

* add directed in API

* comments

* minor function signature fix

* Update csrc/cpu/neighbor_sample_cpu.cpp

* Update csrc/neighbor_sample.cpp
Co-authored-by: Zecheng Zhang <zecheng@kumo.ai>
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* Replace unordered_map with a faster version

* clone recursively repo when testing
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* temporal neighbor sampling adjustments (part2)

* fix

* version up

* formatting

* fix

* reset

* revert

* temporal neighbor sampling adjustments

* typo

* version up

* formatting

* fix

* reset

* revert

* temporal sample

* hetero_neighbor_sample temporal

* api

* remove redundant function

* refactor

* remove catch output

* debug compile

* testing env

* debug

* debug

* debug

* debug

* debug

* revert

* node time data should not be const

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* Update csrc/cpu/neighbor_sample_cpu.cpp
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* temporal template

* compilation fixes
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>

* adjust tensor creation

* update

* torch randint

* update

* fix type

* update

* add test
Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>

* Fixes for build with the new pytorch

* Fixes for build with the new pytorch

* Fixes for build with the new pytorch

* Add `WITH_PYTHON` to conditionally link to Python.

* Only include `Python.h` when WITH_PYTHON is set.

* Avoid including extensions.h as it includes Python.h.

* Better way to include `getpid()`.

* Define `WITH_PYTHON` when building with setup.py.

* Only include Pyinit when building with Python.

* Only include Pyinit when building with Python.