Commit e6f7005

authored

[ILUVATAR_GPU] Add ixattnbkd backend for flash-attention (#1997)

1 parent a7c261b commit e6f7005Copy full SHA for e6f7005

2 files changed

+1547

-834

lines changed

backends/iluvatar_gpu/kernels/cuda_kernels
- flash_attn_grad_kernel.cu
- flash_attn_kernel.cu

2 files changed

+1547

-834

lines changed

Comments

(0)