4 ms·
pytorch already supports AMD with device=cuda already opensourced ROCm/HIP https://github.com/pytorch/pytorch/blob/fb8876069d89aaf27cc9dace075dd36b8e7df4c7/do
by postmeta 2y ago
pytorch already supports AMD with device=cuda
already opensourced ROCm/HIP
https://github.com/pytorch/pytorch/blob/fb8876069d89aaf27cc9dace075dd36b8e7df4c7/docs/source/notes/hip.rst#hip-interfaces-reuse-the-cuda-interfaces https://github.com/pytorch/pytorch/blob/fb8876069d89aaf27cc9...