[FEEDSTOCK REQUEST] Triton

[ce107]

Describe the package you'd like added
OpenAI Triton for PPC64LE especially as there is no way to use it currently on Summit and other remaining AC922 systems.

Describe how this package fits in with the project
Triton is necessary for PyTorch compile to work - any of my attempts to use PyTorch compile on PPC64LE fails.

Additional context
Currently Triton will not build on PPC64LE because the build script downloads a preselected version of LLVM that is x86_64 and more recently ARM64 specific. Triton could also be useful moving forward with P10 MMA if it can learn to create MMA specialized kernels.

The code in question is in python/setup.py (get_llvm_package_info and friends) - and it can be made to point to an alternative location of a prebuilt PPC64LE version of LLVM - of the right level as explained in the Triton build instructions for an alternative LLVM build.

There are a few other places in the code where architecture/os checks happen and a PPC64LE entry would be necessary.