[Autograd] python custom_op implementation on gather_scatter()

fishmingyu / GeoT

GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU

https://arxiv.org/abs/2404.03019

18 stars 1 forks source link

Closed siriuxyu closed 3 months ago