No CPU backend in triton

compiler FX
1

I’m reproducing a test failure in my pull request, https://hud.pytorch.org/pr/pytorch/pytorch/139141#33075326227, a triton failure in CPU backend.
BUt my triton doesn’t have CPU backend, so I couldn’t test it.
Here’s CPU backends my triton supports:
‘nvidia’: Backend(compiler=<class ‘nvi.CUDABackend’>, driver=<class ‘nvi.CudaDriver’>), ‘amd’: Backend(compiler=<class ‘.HIPBackend’>, driver=<class ‘.HIPDriver’>)}

How to reproduce this issue?

2

I found an experimental triton version with CPU backend, GitHub - triton-lang/triton-cpu: An experimental CPU backend for Triton, guess this is the new version, will try this version first.

3

The Triton CPU backend is currently out of tree (not included with default Triton) and should be installed from GitHub - triton-lang/triton-cpu: An experimental CPU backend for Triton