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

Method DoBlasTrmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1669–1677  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1667}
1668
1669bool CUDABlas::DoBlasTrmv(Stream *stream, blas::UpperLower uplo,
1670 blas::Transpose trans, blas::Diagonal diag, uint64 n,
1671 const DeviceMemory<float> &a, int lda,
1672 DeviceMemory<float> *x, int incx) {
1673 return DoBlasInternal<float>(
1674 cublasStrmv, stream, true /* = pointer_mode_host */,
1675 CUDABlasUpperLower(uplo), CUDABlasTranspose(trans),
1676 CUDABlasDiagonal(diag), n, GpuMemory(a), lda, GpuMemoryMutable(x), incx);
1677}
1678
1679bool CUDABlas::DoBlasTrmv(Stream *stream, blas::UpperLower uplo,
1680 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