src/array/cuda/gather_mm.cu · 170203ae2b23ddd8eb05f720298269c20e4b313b · OpenDAS / dgl · GitLab

Find file Blame History Permalink

[Performance] Creating out buffers for `segment_mm`|`sddmm` via `torch.empty()` (#5462) · 170203ae
Xin Yao authored Mar 23, 2023
```
* update for segmentMM

* update for sddmm

* fix a bug
```
170203ae

gather_mm.cu 19 KB