search

rusty1s / pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations
https://pytorch-scatter.readthedocs.io
MIT License
1.5k stars 178 forks source link

[ROCm] fixes ambiguous calls to `shfl*` where there is no explicit type conversion from `c10::Half` to `__half` #411

Closed ashwinma closed 5 months ago

ashwinma commented 5 months ago

Also resolves https://github.com/ROCm/ROCm/issues/2783

Similar PR in pytorch_sparse: https://github.com/rusty1s/pytorch_sparse/pull/360