• Y
    Add a pass to fuse fc+elementwise_add+layernorm (#19776) · 3cd985a6
    Yiqun Liu 提交于
    * Add fc_elementwise_layernorm_fuse pass and unittest.
    
    * Add fused_fc_elementwise_layernorm op and its GPU kernel.
    test=develop
    
    * Apply fc_elementwise_layernorm_fuse_pass to GPU inference.
    
    * Add the setting of attrs in the definition of binary_op.
    test=develop
    
    * Add comment.
    
    * Implement the unittest.
    test=develop
    
    * Change the unittest name of layer_norm.
    test=develop
    3cd985a6
CMakeLists.txt 610 字节