• X
    [ CherryPick ] Cherry pick for einsum optimization. (#43468) · 22e75d92
    xiongkun 提交于
    * [EinsumOp] Polish forward logic and backward logic for optimize (#42603)
    
    * change logic for optimize
    
    * modifty
    
    * merge
    
    * change einsum_v2 as default and add new flags: FLAG_einsum_opt=1|0 (#43010)
    
    * [EinsumOp] Make EinsumOp support bfloat16. (#43085)
    
    * change einsum_v2 as default and add new flags: FLAG_einsum_opt=1|0
    
    * make EInsumOP support bf16
    
    * add unittest for BF16
    
    * add condition for test_BF16
    
    * fix bugs
    
    * fix
    
    * change the backward api to fit einsum op
    22e75d92
einsum_op.cc 3.6 KB