• Z
    Add Multihead matmul fuse pass (#20167) · b8333ede
    zhaoyuchen2018 提交于
    * Add multihead fuse pass for ernie opt
    
    * Refine softmax
    
    test=develop
    
    * Refine cuda kernel
    
    * Refine cuda version
    
    * Refine cmake
    
    test=develop
    
    * refine header file
    
    * refine test case and pass
    * refine comments
    b8333ede
multihead_matmul_fuse_pass.cc 18.1 KB