PyTorch 2.5 ROCm wheels result with two copies of triton ?

[naromero77amd]

My installation recipe involves first install the latest PyTorch wheel. Then a pip install . of causal-conv1d and then a pip install . for mamba ssm.

During the pip install of causal-conv1d, I see this message:

Requirement already satisfied: pytorch-triton-rocm==3.1.0 in /opt/conda/envs/py_3.10/lib/python3.10/site-packages (from torch->causal_conv1d==1.4.0) (3.1.0)

During the pip install of mamba_ssm, I see this message:

Collecting triton (from mamba_ssm==2.2.2)
Downloading triton-3.1.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (1.3 kB)

Is this the correct behavior? Or is one version of Triton for the AMD GPU and the other wheel that is installed for the CPU?

cc: @amoskvic