Skip to content
This repository has been archived by the owner on Jun 22, 2024. It is now read-only.

Add support for AMD GPUs via ROCm #1

Open
ic4-y opened this issue Apr 2, 2024 · 1 comment
Open

Add support for AMD GPUs via ROCm #1

ic4-y opened this issue Apr 2, 2024 · 1 comment

Comments

@ic4-y
Copy link

ic4-y commented Apr 2, 2024

I was wondering if we could add support for AMD GPUs running ROCm in the near future? I could help and could test it on two different AMD cards.

I think in order to get there, we might have to do the following:

  • Provide all the ROCm related packages.

    Probably best to do this in another environment than the default one right now. I was wondering how we avoid duplicating most of the code here? (My Nix greatness is rather limited I have to say)

  • Set some environment variables, so that ROCm knows which GPU architecture we are using. This is particularly important in order to get this running on GPUs that are not officially supported by ROCm, but work nonetheless (which is basically all RDNA3, RDNA2 consumer and workstation GPUs, and also a significant number of RDNA-based GPUs).

    Since users will have different GPUs, is there a (possibly hacky) way to pick this up dynamically from the environment?

@ic4-y
Copy link
Author

ic4-y commented Apr 4, 2024

I think we're getting there...!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant