• H
    Add gpu kernel for new api : linalg.lstsq (#38621) · 405103d8
    Haohongxiang 提交于
    * add lstsq gpu kernel
    
    * update
    
    * add docs_en
    
    * modify ut
    
    * fix bugs
    
    * modify example in docs_en
    
    * remove lstsq_op.cu from ROCM cmake
    
    * modify docs_en
    
    * modify docs_en
    
    * modify docs_en
    
    * remove unneccessary TensorCopy
    405103d8
test_linalg_lstsq_op.py 8.6 KB