• L
    [cherry-pick]Add sparse attention cherrypick (#36447) · 36edb0e1
    Liu-xiandong 提交于
        The code of this PR can only support CUDA 11.2. Currently, CI does not have GPU with CUDA 11.2 , and all tests will be skipped automatically.
    
        The new OP is paddle._C_ops.sparse_attention. Regarding the work of the python API, it will be resolved in a follow-up PR.
    
        The code of this PR lacks tests on dynamic graphs and static graphs, and will be added in subsequent PRs.
    36edb0e1
CMakeLists.txt 8.7 KB