• G
    Add cholesky_op (#23543) · a8c0fb4e
    Guo Sheng 提交于
    * Add cholesky_op forward part. test=develop
    
    * Complete cholesky_op forward part. test=develop
    
    * Add cholesky_op backward part. test=develop
    
    * Complete cholesky_op backward part. test=develop
    
    * Refine cholesky_op error check and docs. test=develop
    
    * Add grad_check unit test for cholesky_op. test=develop
    
    * Fix sample code in cholesky doc. test=develop
    
    * Refine some error messages of cholesky_op. test=develop
    
    * Refine some error messages of cholesky_op. test=develop
    
    * Remove unused input in cholesky_grad. test=develop
    
    * Remove unused input in cholesky_grad. test=develop
    
    * Fix stream for cusolverDnSetStream. test=develop
    
    * Update PADDLE_ENFORCE_CUDA_SUCCESS from cholesky_op to adapt to latest code.
    test=develop
    
    * Add CUSOLVER ERROR in enforce.h
    test=develop
    
    * Fix the missing return value in cholesky. test=develop
    a8c0fb4e
cholesky_op.h 13.9 KB