Cannot run the tutorial example "06-fused-attention.py"

[rozyangno]

import triton.tools.experimental_descriptor
ModuleNotFoundError: No module named 'triton.tools.experimental_descriptor'

Triton is good. But tutorial examples seem obsolete. They are important for new comers. Can they be updated?

[albertgong1]

I also encountered a similar issue when running this version of the tutorial. My setup is the following:

• python==3.12
• triton==3.3.1
• A6000 with 48 GB memory
• Driver Version: 530.30.02
• CUDA Version: 12.1

I believe the issue is that the latest tutorial no longer supports Ampere GPUs and older, since it relies on TMA hardware support, which was only introduced on H100. Running an older version of the tutorial worked for me: https://github.com/triton-lang/triton/blob/8bd8035559d4d2fa387149eecdf10d545d8a3d03/python/tutorials/06-fused-attention.py