Xiangxi Guo (Ryan) f0d5d0111f Avoid torch compile graphbreak for older pytorch versions (#9344) ...

Turns out torch.compile has some gaps in context manager decorator
syntax support. I've sent patches to fix that in PyTorch, but it won't
be available for all the folks running older versions of PyTorch, hence
this trivial patch.

2025-08-14 23:41:37 -04:00

18 KiB

Raw Blame History

View Raw