• L
    Update elementwise double grad to save gpu memory (#19509) · 982e61f5
    Leo Chen 提交于
    * update elementwise double grad to save gpu memory, test=develop
    
    * update elementwise_mul/div_grad_grad to save memory, test=develop
    
    * remove eval function in eigen statement to save memory, test=develop
    
    * add unittest for elementwise_div_grad_grad without dout, test=develop
    
    * add unittest for elementwise_add_grad_grad without ddx, test=develop
    
    * add float16 cuda kernel for elementwise double grad op, test=develop
    982e61f5
elementwise_mul_op.cc 4.2 KB