Aidyn-A a99536480d [ATen][Native][Special] Hermite polynomial prematurely return NaN if n is high (#141955) ... Hermite polynomials diverge to NaN at high orders due to numerical overflow. The proposal is to prematurely return NaN of it is known that at this value it will be NaN. According to my short test ```Python import torch device = "cuda" dtype = torch.float32 x = torch.linspace(-1000, 1000, 100000, device=device, dtype=dtype) for n in range(1024): if torch.special.hermite_polynomial_h(x, n).isnan().sum().item() == x.shape[0]: print(f"hermite_polynomial_h: all outputs are nans! n = {n}") break for n in range(1024): if torch.special.hermite_polynomial_he(x, n).isnan().sum().item() == x.shape[0]: print(f"hermite_polynomial_he: all outputs are nans! n = {n}") break ``` The output values become NaNs at these orders: ``` hermite_polynomial_h: all outputs are nans! n = 53, dtype=torch.float32 hermite_polynomial_he: all outputs are nans! n = 61, dtype=torch.float32 hermite_polynomial_h: all outputs are nans! n = 272, dtype=torch.float64 hermite_polynomial_he: all outputs are nans! n = 304, dtype=torch.float64 ``` Surely, it makes sense to increase the limit as a safety margin. Pull Request resolved: https://github.com/pytorch/pytorch/pull/141955 Approved by: https://github.com/malfet, https://github.com/eqy		2024-12-18 08:30:08 +00:00
..
__init__.py	[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771)	2024-08-01 17:07:14 +00:00
_masked.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
fft.py	[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771)	2024-08-01 17:07:14 +00:00
linalg.py	[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771)	2024-08-01 17:07:14 +00:00
nested.py	Fix NJT backward tests (#143072)	2024-12-12 18:06:23 +00:00
signal.py	[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771)	2024-08-01 17:07:14 +00:00
sparse.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
special.py	[ATen][Native][Special] Hermite polynomial prematurely return NaN if n is high (#141955)	2024-12-18 08:30:08 +00:00