Results
1
issues of
Lixin Liu
Dear authors, I find that if we use torch_scatter in CUDA graph. The output memory should be pre-allocated (the param `out`) in `scatter_mean`, `scatter_max`, `scatter_add`, etc. Meanwhile, since `scatter_softmax` does...
enhancement