• Z
    Add a Sparse Op: to_sparse_csr (#39333) · 76d527e1
    zhangkaihuo 提交于
    * implement AllocateFrom
    
    * dense_to_sparse_coo
    
    * optimize unit testing; support rocm
    
    * 1. delete fluid related header file
    2. update the copyright
    
    * fix hipMemcpy
    
    * update dense_to_sparsecoo
    
    * add namespace sparse
    
    * sparse_csr_to_dense
    
    * test to_sparse_coo: csr_to_coo
    
    * fix writing error
    
    * to_sparse_csr: dense_to_sparse_csr and sparse_coo_to_csr
    
    * fix check shape
    
    * fix unit test
    
    * replace CUDADeviceContext by GPUContext
    76d527e1
sparse_api.cc 6.1 KB