[Feature]: (FIX) `triton` should not be installed on `ppc64le`

[Shafi-Hussain]

🚀 The feature, motivation and pitch

triton==3.2.0 was added to requirements/cpu.txt in #16384

Triton is a GPU specific requirement and should not be added to CPU specific requirements file.
Request to move it to requirements/cuda.txt

Refer: [Triton] - language/compiler for specifying and compiling tile programs into efficient GPU code.
https://www.eecs.harvard.edu/~htk/publication/2019-mapl-tillet-kung-cox.pdf

Alternatives

No response

Additional context

This is breaking ppc64le cpu builds of vLLM

Before submitting a new issue...

• Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

[hmellor]

Triton is a torch dependency that was being installed anyway before that PR. It was added as a hard dependency in that PR to prevent a transient issue with triton 3.3.0 being installed in the Helm test in GitHub actions which uses the vLLM for CPU.

See https://vllm-dev.slack.com/archives/C07QP347J4D/p1744243455177579 for the full context.

[Shafi-Hussain]

@hmellor torch already has it pinned to version 3.2.0 (in torch 2.6.0) and is installed only for x86. Maybe we can have a conditional check to have it installed only for x86 architecture then?

[Akashcodes732]

Hi @hmellor , this breaks the build for ppc64le as triton is not supported for IBM POWER.

[mkumatag]

Hi @hmellor , this breaks the build for ppc64le as triton is not supported for IBM POWER.

can we just ignore this package for ppc64le platform then?

[hmellor]

can we just ignore this package for ppc64le platform then?

I've made a PR #16470 to do this