MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / DoBlasTpmv

Method DoBlasTpmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1585–1593  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1583}
1584
1585bool CUDABlas::DoBlasTpmv(Stream *stream, blas::UpperLower uplo,
1586 blas::Transpose trans, blas::Diagonal diag, uint64 n,
1587 const DeviceMemory<float> &ap, DeviceMemory<float> *x,
1588 int incx) {
1589 return DoBlasInternal<float>(
1590 cublasStpmv, stream, true /* = pointer_mode_host */,
1591 CUDABlasUpperLower(uplo), CUDABlasTranspose(trans),
1592 CUDABlasDiagonal(diag), n, GpuMemory(ap), GpuMemoryMutable(x), incx);
1593}
1594
1595bool CUDABlas::DoBlasTpmv(Stream *stream, blas::UpperLower uplo,
1596 blas::Transpose trans, blas::Diagonal diag, uint64 n,

Callers

nothing calls this directly

Calls 6

CUDABlasTransposeFunction · 0.85
CUDABlasDiagonalFunction · 0.85
GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
GpuComplexFunction · 0.85
CUDABlasUpperLowerFunction · 0.70

Tested by

no test coverage detected