Transformers.jl icon indicating copy to clipboard operation
Transformers.jl copied to clipboard

Published 20 hours ago • verified publisher icon chengchingwen

AMDGPU support

Open radudiaconu0 opened this issue 1 year ago • 1 comments

Can we have amdgpu support?

radudiaconu0 avatar Nov 22 '23 10:11 radudiaconu0

Shouldn't be hard to add. Generally it just needs to make sure a few functions, like NeuralAttentionlib's gemm, dispatch to the correct functions. But I don't have amdgpu to test.

chengchingwen avatar Nov 23 '23 02:11 chengchingwen