RWKV-LM icon indicating copy to clipboard operation
RWKV-LM copied to clipboard

Published 20 hours ago • verified publisher icon BlinkDL

Radeon Open Compute support

Open erkinalp opened this issue 2 years ago • 1 comments

There are many users with AMD graphics cards that want to train this model in a GPU accelerated manner. Radeon Open Compute is AMD's equivalent to CUDA (the relevant component in Radeon Open Compute is called HIP).

  • [ ] Attempt to HIPify the codebase
  • [ ] Make the HIP port optimised for full performance

erkinalp avatar Jan 21 '23 19:01 erkinalp

Cool. Let me know if you will work on it

BlinkDL avatar Jan 24 '23 08:01 BlinkDL

Here is AMD Ubuntu 22.04 GPU guide https://github.com/BlinkDL/ChatRWKV/issues/15

BlinkDL avatar Feb 26 '23 17:02 BlinkDL

Cool, waiting for it.

linwk20 avatar Jun 13 '23 14:06 linwk20

subtask 2 is not complete, should be reopened

erkinalp avatar Jun 13 '23 14:06 erkinalp