• 0
    Add matrix_rank Op and it's GPU and CPU kernel (#34823) · eb1fbf12
    0x45f 提交于
    * init matrix_rank op, add matrix_rank CPU code and test
    
    * add GPU kernel, remove svd_eigen.h
    
    * add CPU kernel when tol is tensor
    
    * add cpu and gpu code when tol is tensor
    
    * fix CI-ROCM error
    
    * add matrix_rank API describe, fix PR-CI-Py3 error
    
    * fix PR-CI-Windows error, add matrix_rank API test
    
    * delete useless comments
    
    * fix review
    
    * add my code in svd_helper.h
    
    * update doc commets
    
    * remove spaces
    eb1fbf12
matrix_rank_op.cu 13.4 KB