Sarmiento Fernando

Results 2 issues of Sarmiento Fernando

I tried to replicate the results from your benchmarks using docker gpu with this docker image nvidia/cuda:12.6.1-cudnn-devel-ubuntu22.04 and nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04. After I install all the requirements I cannot get close to...

It will be great to return the index locations from the scatter operations like torch_scatter: `from torch_scatter import scatter_max src = torch.Tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3,...