Fix RWKV backward on GPU
# What does this PR do?

Fixes the backward pass for RWKV on GPU. The function backward was not adapted to the revamp of the forward, my bad.

_The documentation is not available anymore as the PR was closed or merged._