News for hackers. Stuff that matters.

A comment on Swift for TensorFlow – A system for deep learning and differentiable computing. Parent comment

Re:Swift for TensorFlow – A system for deep learning and dif... by c16a on Sunday September 20, 2020 @03:02PM (#24536421)
Tensorflow can work with ROCM which is equivalent to Nvidia's CUDA.
[ Parent | on HN ]
Re:Swift for TensorFlow – A system for deep learning and dif... by yunohn on Monday September 21, 2020 @04:22PM (#24548017)
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.
[ Parent | on HN ]
Re:Swift for TensorFlow – A system for deep learning and dif... by amkkma on Monday September 21, 2020 @05:04PM (#24548463)
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

[ Parent | on HN ]
Re:Swift for TensorFlow – A system for deep learning and dif... by yunohn on Monday September 21, 2020 @05:46PM (#24548829)
Indeed, the Julia scene looks enticing right now. If only AMD ROCM supported MacOS... :/
[ Parent | on HN ]