RingFlashAttnFunc forward backward arguments mismatch #149
Description
When using RingFlashAttnFunc I get the error that the backward pass is expected to return 14 gradients (as the corresponding forward pass takes 14 arguments besides ctx) but it actually only returns 13 values. To fix this, one additional None should be added to the backward function of RingFlashAttnFunc.