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

Method DoBlasTbsv

tensorflow/stream_executor/cuda/cuda_blas.cc:1539–1548  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1537}
1538
1539bool CUDABlas::DoBlasTbsv(Stream *stream, blas::UpperLower uplo,
1540 blas::Transpose trans, blas::Diagonal diag, uint64 n,
1541 uint64 k, const DeviceMemory<float> &a, int lda,
1542 DeviceMemory<float> *x, int incx) {
1543 return DoBlasInternal<float>(
1544 cublasStbsv, stream, true /* = pointer_mode_host */,
1545 CUDABlasUpperLower(uplo), CUDABlasTranspose(trans),
1546 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, GpuMemoryMutable(x),
1547 incx);
1548}
1549
1550bool CUDABlas::DoBlasTbsv(Stream *stream, blas::UpperLower uplo,
1551 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