Torch.compile support for Python 3.12 completed

Signal boosting that Python 3.12 support has been added to torch.compile and has been present in the nightly builds for a while. We anticipate that this feature will be included in the PyTorch 2.4 release.

We encountered segfaults with certain older versions of triton while working on 3.12 support - if you encounter one, try to build triton from source on the latest main commit.

Please give torch.compile a try on 3.12 and report any issues to the PyTorch GitHub!

5 Likes