• J
    【Prim】Fix dropout CINN amp error (#51688) · 94cd1ba2
    Jiabin Yang 提交于
    * support amp logic for layer_norm and softmax
    
    * fix layer_norm amp
    
    * fix layernorm api and dropout fp16
    
    * fix layernorm api and dropout fp16
    
    * fix bn, ln dtype in float16
    
    * fix dropout fp16
    
    * fix comment
    
    * fix cinn dropout amp error
    94cd1ba2
composite_rules.py 14.2 KB