torchdiffeq icon indicating copy to clipboard operation
torchdiffeq copied to clipboard
verified publisher icon rtqichen

torch.export-friendly data-dependent assertions in misc.py, solvers.py

Open pianpwk opened this issue 7 months ago • 1 comments

Ran into these errors while trying to export F5-TTS, rewriting to torch._check calls to reduce data-dependent errors, graph breaks in torch.compile

pianpwk avatar May 22 '25 01:05 pianpwk

@rtqichen Bumping this. I am also running into similar issues with torch export, enabling this would be very useful

aw632 avatar Jun 10 '25 22:06 aw632