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	4987	February 15, 2023
I managed to make Flux.jl work with GPUs again! Machine Learning flux	6	4396	February 18, 2020
AMDGPU available for Windows? GPU	4	1408	February 18, 2021
Can you \|> gpu a GraphConv in GeometricFlux.jl? Machine Learning question , package , flux	0	332	March 26, 2021
Flux: GPU not working as expected Machine Learning flux	6	2177	July 28, 2020