예제 #1
0
파일: csr.py 프로젝트: Hulalazz/cupy
 def eliminate_zeros(self):
     """Removes zero entories in place."""
     if cupy.cuda.runtime.runtimeGetVersion() >= 8000:
         compress = cusparse.csr2csr_compress(self, 0)
     else:
         coo = self.tocoo()
         coo.eliminate_zeros()
         compress = coo.tocsr()
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr
예제 #2
0
 def eliminate_zeros(self):
     """Removes zero entories in place."""
     if cupy.cuda.runtime.runtimeGetVersion() >= 8000:
         compress = cusparse.csr2csr_compress(self, 0)
     else:
         coo = self.tocoo()
         coo.eliminate_zeros()
         # Because tocsr sums duplicated entries, it cannot keep nnz
         compress = coo.tocsr()
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr
예제 #3
0
파일: csr.py 프로젝트: mnicely/cupy
 def eliminate_zeros(self):
     """Removes zero entories in place."""
     compress = cusparse.csr2csr_compress(self, 0)
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr