Hacker News new | ask | show | jobs
by c16a 2103 days ago
Tensorflow can work with ROCM which is equivalent to Nvidia's CUDA.
1 comments
yunohn 2101 days ago
There is no official support, only AMD's custom TF branch. Further ROCM itself works only on Linux. I believe most people writing Swift are using MacOS.
link
amkkma 2101 days ago
And Julia can do custom ROCM codegen, either by itself or through array and kernel abstractions :) https://github.com/JuliaGPU/AMDGPU.jl

https://juliagpu.gitlab.io/KernelAbstractions.jl/

https://github.com/JuliaGPU/GPUArrays.jl/pulse

link
yunohn 2101 days ago
Indeed, the Julia scene looks enticing right now. If only AMD ROCM supported MacOS... :/
link