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

Method DoBlasTrsv

tensorflow/stream_executor/cuda/cuda_blas.cc:1711–1719  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1709}
1710
1711bool CUDABlas::DoBlasTrsv(Stream *stream, blas::UpperLower uplo,
1712 blas::Transpose trans, blas::Diagonal diag, uint64 n,
1713 const DeviceMemory<float> &a, int lda,
1714 DeviceMemory<float> *x, int incx) {
1715 return DoBlasInternal<float>(
1716 cublasStrsv, stream, true /* = pointer_mode_host */,
1717 CUDABlasUpperLower(uplo), CUDABlasTranspose(trans),
1718 CUDABlasDiagonal(diag), n, GpuMemory(a), lda, GpuMemoryMutable(x), incx);
1719}
1720
1721bool CUDABlas::DoBlasTrsv(Stream *stream, blas::UpperLower uplo,
1722 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