4 ms·

Tensorflow can work with ROCM which is equivalent to Nvidia's CUDA.

• yunohn 6y 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.
  • amkkma 6y ago
    And Julia can do custom ROCM codegen, either by itself or through array and kernel abstractions :) https://github.com/JuliaGPU/AMDGPU.jl https://github.com/JuliaGPU/AMDGPU.jl https://juliagpu.gitlab.io/KernelAbstractions.jl/ https://juliagpu.gitlab.io/KernelAbstractions.jl/ https://github.com/JuliaGPU/GPUArrays.jl/pulse https://github.com/JuliaGPU/GPUArrays.jl/pulse
    • yunohn 6y ago
      Indeed, the Julia scene looks enticing right now. If only AMD ROCM supported MacOS... :/