3 ms·
Pretty cool, and it seems to work on AMD GPUs as well. I've just tried it on my RX6800 with the ROCm build of PyTorch.
by Gazoche 4y ago
Pretty cool, and it seems to work on AMD GPUs as well. I've just tried it on my RX6800 with the ROCm build of PyTorch.