Use `.scalar_type()` in `AT_DISPATCH`

lab-cosmo / sphericart

Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

https://sphericart.readthedocs.io/en/latest/

MIT License

73 stars 13 forks source link

Closed malfet closed 3 weeks ago

malfet commented 3 weeks ago

As AT_DISPATCH_FLOATING_TYPES(x.type() pattern is deprecated since https://github.com/pytorch/pytorch/pull/17996

jermenkoo commented 3 weeks ago

Beat me to it by a few minutes! :)

Luthaf commented 3 weeks ago

If I may @malfet, could you give a quick look at https://github.com/pytorch/pytorch/pull/131340, and/or recommend a different reviewer?