Make Composer compatible with Python 3.12.

[antoinebrl]

🚀 Motivation

Since 2.4 released ~3 months ago PyTorch is officially supporting Python 3.12 (the torch.compile was a bit lagging behind due to changes in the CPython interface). It would be great to see Py3.12 being supported in Composer too.

[mvpatel2000]

Will likely be in next release :)

[b-chu]

We're seeing difficulties with upgrading to Python 3.12 with torchvision dependencies so will hold off for this release.
https://github.com/mosaicml/composer/actions/runs/11414612796

[antoinebrl]

Thanks @b-chu for having a look and reporting on your investigation! 👍 Should this issue to reopened to track the adoption of py3.12 ?

[mvpatel2000]

Concretely, we are blocked on upgrading until triton fixes some issues.

[antoinebrl]

Have the issues been raised upstream ?

[b-chu]

Hi @antoinebrl, we haven't root caused the triton issues. We'll be doing a release soon so hopefully the Python 3.12 issues have been fixed, but if not, we'll investigate more. Thanks!