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

Method DoBlasTbmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1493–1502  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1491}
1492
1493bool CUDABlas::DoBlasTbmv(Stream *stream, blas::UpperLower uplo,
1494 blas::Transpose trans, blas::Diagonal diag, uint64 n,
1495 uint64 k, const DeviceMemory<float> &a, int lda,
1496 DeviceMemory<float> *x, int incx) {
1497 return DoBlasInternal<float>(
1498 cublasStbmv, stream, true /* = pointer_mode_host */,
1499 CUDABlasUpperLower(uplo), CUDABlasTranspose(trans),
1500 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, GpuMemoryMutable(x),
1501 incx);
1502}
1503
1504bool CUDABlas::DoBlasTbmv(Stream *stream, blas::UpperLower uplo,
1505 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