Commits · 01dc5042eee16fddadc0ec5aff6a5b0747196e71 · OpenDAS / torch-sparse

19 Aug, 2022 1 commit

Add `spmm` bf16 support (#269) · 01dc5042

YanbingJiang authored Aug 19, 2022



* 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>

01dc5042

18 Aug, 2022 1 commit
- update (#268) · f7c74ec0
  Matthias Fey authored Aug 18, 2022
  
  f7c74ec0
17 Aug, 2022 1 commit
- fix building of wheels · ac7f314c
  rusty1s authored Aug 17, 2022
  
  ac7f314c
11 Aug, 2022 1 commit
- use batch idx and node id as unique key for dedup in temporal sampling (#267) · 6143af21
  Dong Wang authored Aug 11, 2022
```
Co-authored-by: Dong Wang <d@dongs-mbp.lan>
```
  6143af21
09 Aug, 2022 1 commit
- Replace unordered_map with phmap in hetero_sample (#266) · 916ba55b
  OlhaBabicheva authored Aug 09, 2022
  
  916ba55b
06 Aug, 2022 1 commit
- make graph sample deterministic when iterate different type (#265) · ae22058a
  Dong Wang authored Aug 05, 2022
```
Co-authored-by: Dong Wang <d@dongs-mbp.lan>
```
  ae22058a
30 Jul, 2022 1 commit
- version up · 7f6379aa
  rusty1s authored Jul 30, 2022
  
  7f6379aa
28 Jul, 2022 1 commit
- init (#263) · 03f2caa9
  Manan Shah authored Jul 27, 2022
  
  03f2caa9
23 Jul, 2022 1 commit
- import `torch/torch.h` instead of `torch/extension.h` (#260) · 27633e2e
  Gerico Vidanes authored Jul 23, 2022
  
  27633e2e
16 Jul, 2022 2 commits

Minor refactor for temporal sampling (#257) · ad228bb8

Rex Ying authored Jul 16, 2022



* 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>

ad228bb8

Temporal sample disable undirected (#247) · d670cddb

Rex Ying authored Jul 15, 2022



* 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>

d670cddb

14 Jul, 2022 2 commits
- Replace unordered_map with a faster version (#254) · 7dbc51cd
  kgajdamo authored Jul 14, 2022
```
* Replace unordered_map with a faster version

* clone recursively repo when testing
Co-authored-by: Matthias Fey <matthias.fey@tu-dortmund.de>
```
  7dbc51cd
- fix metis (#255) · 63b75d8d
  Matthias Fey authored Jul 14, 2022
  
  63b75d8d
13 Jul, 2022 1 commit
- Updated README to reflect conda limitations (#253) · bd7273e2
  Anil Radhakrishnan authored Jul 13, 2022
  
  bd7273e2
30 Jun, 2022 3 commits
- fix conda build [ci skip] · 8cdb5282
  rusty1s authored Jun 30, 2022
  
  8cdb5282
- Version up (#246) · 587d1d02
  Matthias Fey authored Jun 30, 2022
```
* PyTorch 1.12 support

* version up
```
  587d1d02
- PyTorch 1.12 support (#245) · 897c0a8d
  Matthias Fey authored Jun 30, 2022
  
  897c0a8d
25 Jun, 2022 1 commit
- add doc · 08892a31
  rusty1s authored Jun 25, 2022
  
  08892a31
18 May, 2022 1 commit
- add all headers · adc8be24
  rusty1s authored May 18, 2022
  
  adc8be24
26 Apr, 2022 1 commit
- out of range exception catching (#227) · 79cffd57
  Rex Ying authored Apr 25, 2022
  
  79cffd57
25 Apr, 2022 2 commits
- Temporal neighbor sampling adjustments (part2) (#226) · 0172aeb3
  Matthias Fey authored Apr 25, 2022
```
* temporal neighbor sampling adjustments (part2)

* fix
```
  0172aeb3
- Temporal `neighbor_sample` adjustments (#225) · caf7ddde
  Matthias Fey authored Apr 25, 2022
```
* version up

* formatting

* fix

* reset

* revert

* temporal neighbor sampling adjustments

* typo
```
  caf7ddde
22 Apr, 2022 2 commits

Version up (#224) · 15c56351
Matthias Fey authored Apr 22, 2022
```
* version up

* formatting

* fix

* reset

* revert
```
15c56351

Temporal sampling (#202) · 79535f30

Rex Ying authored Apr 22, 2022



* 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>

79535f30

20 Apr, 2022 1 commit
- Fix tensor creation (#222) · eafcfe0a
  Matthias Fey authored Apr 20, 2022
```
* adjust tensor creation

* update
```
  eafcfe0a
16 Apr, 2022 1 commit
- used explicit at::Half to __half conversion operator to fix ambiguous conversion error (#220) · 7a6c9ab4
  Alessandro Pacchioni authored Apr 16, 2022
```
Co-authored-by: Alessandro Pacchioni <apacchioni@ambarella.com>
```
  7a6c9ab4
02 Apr, 2022 1 commit

`rand()` -> `torch::randint()` (#217) · 122811a6

Zeyuan Tan authored Apr 02, 2022



* torch randint

* update

* fix type

* update

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

122811a6

25 Mar, 2022 1 commit
- fix cuda context · 124bc097
  rusty1s authored Mar 25, 2022
  
  124bc097
22 Mar, 2022 1 commit
- CMake fixes (#213) · 2da6195d
  Matthias Fey authored Mar 22, 2022
```
* cmake fixes

* update

* update

* update

* update

* reset
```
  2da6195d
21 Mar, 2022 3 commits
- cmake fixes (#212) · 66e70ae4
  Matthias Fey authored Mar 21, 2022
  
  66e70ae4
- update readme [ci skip] · 3c64bb0d
  rusty1s authored Mar 21, 2022
  
  3c64bb0d
- Add tests for neighbor sampling (#210) · 723039b8
  Padarn Wilson authored Mar 21, 2022
```
* fixing neighbour sampling when repeated nodes

* fix problem with indexing

* reset

* reset

* reset

* add tests
Co-authored-by: rusty1s <matthias.fey@tu-dortmund.de>
```
  723039b8
12 Mar, 2022 1 commit
- fix test · b38d0b5e
  rusty1s authored Mar 12, 2022
  
  b38d0b5e
11 Mar, 2022 6 commits
- [ci skip] · c4f318ee
  rusty1s authored Mar 11, 2022
  
  c4f318ee
- [ci skip] · 09a90659
  rusty1s authored Mar 11, 2022
  
  09a90659
- torch_csr_tensor · b0ff709e
  rusty1s authored Mar 11, 2022
  
  b0ff709e
- skip metis test · fcf15650
  rusty1s authored Mar 11, 2022
  
  fcf15650
- fix testing · 80fe5c0f
  rusty1s authored Mar 11, 2022
  
  80fe5c0f
- version up + PyTorch 1.11 support · f79c8e98
  rusty1s authored Mar 11, 2022
  
  f79c8e98
09 Feb, 2022 1 commit

Fix issue 193 (#203) · 3ec1eacf

Michał Marcinkiewicz authored Feb 09, 2022

* Fixes for build with the new pytorch

* Fixes for build with the new pytorch

* Fixes for build with the new pytorch

3ec1eacf