AMD gpu update

[stevelcb]

Ubuntu 22.04

Hi everyone I thought I'd update on this having tried to get ROCm gpu acceleration recognised via onnx.

We created the environment for building GraX as here: https://github.com/Steffenhir/GraXpert

Then activated AMD's ROCm as here: https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-onnx.html

That works fine and AMD's ROCm is indeed available via onnxruntime:

>>> import onnxruntime as ort
>>> ort.get_available_providers()
['MIGraphXExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']

We then build... However, GraX still sees only the CPU:

2024-06-15 12:45:13,787 MainProcess root INFO     Starting denoising
2024-06-15 12:45:15,548 MainProcess root INFO     Available inference providers : ['CPUExecutionProvider']
2024-06-15 12:45:15,548 MainProcess root INFO     Used inference providers : ['CPUExecutionProvider']
2024-06-15 12:45:17,962 MainProcess root INFO     Progress: 1%
2024-06-15 12:45:19,893 MainProcess root INFO     Progress: 2%

Reading to the end of the AMD document, I see that it works with: Radeon: RX 7900 XTX, RX 7900 XT, RX 7900, GRE PRO W7900 and PRO W7800

I have a gfx90, so not sure if the gpu will be visible to GraX. It is to other programs, such as StarTools but that's via opencl.

Still thinking... Any ideas anyone? Cheers and TIA