(ctx, grad_output)
| 236 | |
| 237 | @staticmethod |
| 238 | def backward(ctx, grad_output): |
| 239 | theta, = ctx.saved_tensors |
| 240 | _apply_rotary_emb_inplace(grad_output, theta, conj=not ctx.conj) |
| 241 | return grad_output, None, None |
| 242 | |
| 243 | |
| 244 | def apply_rotary_emb_(x, theta): |
no test coverage detected