• C
    Integration flash attention (#49869) · 61611786
    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
    61611786
test_flash_attention.py 5.9 KB