This commit is contained in:
Haoming 2026-05-26 12:57:21 +08:00
parent d6ec03ec30
commit f4a9f9c262

View File

@ -55,8 +55,8 @@ def without_autocast(*args, **kwargs):
return contextlib.nullcontext()
# class NansException(Exception):
# pass
class NansException(Exception):
pass
def test_for_nans(x: torch.Tensor, *args, **kwargs):