pytorch Architecture Docs
Home / Class/ csrgeam2Nnz Class — pytorch Architecture

csrgeam2Nnz Class — pytorch Architecture

Architecture documentation for the csrgeam2Nnz class in CUDASparseBlas.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/cuda/CUDASparseBlas.h lines 58–76

template <typename scalar_t>
inline void csrgeam2Nnz(CUSPARSE_CSRGEAM2_NNZ_ARGTYPES()) {
  TORCH_CUDASPARSE_CHECK(cusparseXcsrgeam2Nnz(
      handle,
      m,
      n,
      descrA,
      nnzA,
      csrSortedRowPtrA,
      csrSortedColIndA,
      descrB,
      nnzB,
      csrSortedRowPtrB,
      csrSortedColIndB,
      descrC,
      csrSortedRowPtrC,
      nnzTotalDevHostPtr,
      workspace));
}

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free