- 15 8月, 2023 1 次提交
-
-
由 yinwei 提交于
--------- Co-authored-by: Ntianhaodongbd <tianhaodong@baidu.com>
-
- 09 8月, 2023 1 次提交
-
-
由 umiswing 提交于
-
- 07 8月, 2023 1 次提交
-
-
由 yin wei 提交于
* add mask * add backword * add enforce info * update scale * integrate code * update enforce * add enforce eq * add error type * update enforce * add test_flash_attention * Polish codes and fix compiling errors. * Set num_splits to 0 for flash-attn with tensor mask. * Fix the compiling error for non flash-attn case. --------- Co-authored-by: NLiu Yiqun <liuyiqun01@baidu.com>
-
- 02 8月, 2023 1 次提交
-
-
由 zhenhailiu 提交于
-
- 14 6月, 2023 1 次提交
-
-
由 Yuang Liu 提交于
-
- 12 6月, 2023 1 次提交
-
-
由 Nyakku Shigure 提交于
-
- 05 5月, 2023 1 次提交
-
-
由 zhangkaihuo 提交于
-
- 20 4月, 2023 1 次提交
-
-
由 Chitsing KUI 提交于
* add flash randomness control * fix VLOG undefied
-
- 06 4月, 2023 1 次提交
-
-
由 sneaxiy 提交于
* fix flash attn * fix another API
-
- 16 3月, 2023 1 次提交
-
-
由 Chitsing KUI 提交于
* rename flash_attn_raw to flash_attn_unpadded * fix static api * fix static return
-
- 01 3月, 2023 1 次提交
-
-
由 Chitsing KUI 提交于
* flash attn * seed * almost * softmax * fix workspace * add unitest; linux only * fix setup * fix datatype include * fix setup typo * fix def scope * new error api * use paddle fork * fix attr bug; complete ut * update flash hash * fix rng reset * fix offset * fix comments
-