Instead of replacing all #ifdef USE_CUDA, just add USE_CUDA define to ROCm build.
Attach a file by drag & drop or click to upload