pytorch/c10/metal
Davide Italiano 91c4bf39d3 [mps] Add a shader for spherical_bessel_j0. (#146771)
In preparation for adding the operation to inductor/eager.
Adapted from the CUDA version of the shader.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/146771
Approved by: https://github.com/malfet
2025-02-09 05:11:17 +00:00
..
random.h [MPSInductor] Add rand support (#145705) 2025-01-27 06:07:36 +00:00
reduction_utils.h [MPSInductor] Implement Welford reduction (#146703) 2025-02-08 05:00:00 +00:00
special_math.h [mps] Add a shader for spherical_bessel_j0. (#146771) 2025-02-09 05:11:17 +00:00
utils.h [METAL] inline bfloat min/max (#146588) 2025-02-06 17:57:31 +00:00