Add example for AMD GPUs in documentation #1973
Conversation
| Reactant.set_default_backend("gpu") | ||
| ``` | ||
|
|
||
| ```julia [AMD GPUs] |
There was a problem hiding this comment.
How about putting this together with Nvidia GPUs for the time being? The code to execute is just the same.
There was a problem hiding this comment.
yeah so one thing I'm debating doing here, is we could have a commented out line set_default_backend("cuda") vs "rocm" which will enforce which one
There was a problem hiding this comment.
I like being explicit, it reduces confusion.
Maybe should keep "gpu" as an alias of "cuda" for the rest of v0.2 though?
There was a problem hiding this comment.
So the current setup at least has gpu essentially as an Alia's for whatever GPU was found on the system, including when metal earlier worked.
I think it's nice to have both so you can either just say get me a GPU or alternatively make sure you get this specifically
There was a problem hiding this comment.
Looks better (although the code commented out is easy to ignore/miss)
Added comments for setting GPU backends in Reactant usage examples.
No description provided.