• S
    Support FP16 for more ops (#38123) · 1f445bf3
    sneaxiy 提交于
    * support FP16 for more ops
    
    * add amp list tests
    
    * refine reduce_mean_grad
    
    * fix OP benchmark ci
    
    * fix fp16 reduce_mean
    
    * updat ut, but still have some problems
    
    * remove mean/reduce_mean fp16 kernel
    1f445bf3
elementwise_min_op.cu 3.4 KB