PaddleCustomDevice icon indicating copy to clipboard operation
PaddleCustomDevice copied to clipboard
verified publisher icon PaddlePaddle

[MLU] optimize range kernel; flash_attn kernel

Open ShawnNew opened this issue 1 year ago • 1 comments

  1. make range kernel a function, which uses cnnlArange_v2
  2. use range function in flash_attn

ShawnNew avatar Apr 29 '24 08:04 ShawnNew

Thanks for your contribution!

paddle-bot[bot] avatar Apr 29 '24 08:04 paddle-bot[bot]