Porting cuda example to rocm amdgpu

jcbritobr June 2, 2025, 12:35am 1

Hello, good evening.

May someone help to show me how to start porting this example to use amdgpu.jl?

Topic		Replies	Views
Flux with AMD GPU(s)? Machine Learning flux , amdgpu	34	5034	February 15, 2023
I managed to make Flux.jl work with GPUs again! Machine Learning flux	6	4404	February 18, 2020
AMDGPU available for Windows? GPU	4	1409	February 18, 2021
Can you \|> gpu a GraphConv in GeometricFlux.jl? Machine Learning question , package , flux	0	333	March 26, 2021
Flux: GPU not working as expected Machine Learning flux	6	2186	July 28, 2020