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

Method DoBlasTrsm

tensorflow/stream_executor/cuda/cuda_blas.cc:3114–3124  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3112}
3113
3114bool CUDABlas::DoBlasTrsm(Stream *stream, blas::Side side,
3115 blas::UpperLower uplo, blas::Transpose transa,
3116 blas::Diagonal diag, uint64 m, uint64 n, float alpha,
3117 const DeviceMemory<float> &a, int lda,
3118 DeviceMemory<float> *b, int ldb) {
3119 return DoBlasInternal<float>(
3120 cublasStrsm, stream, true /* = pointer_mode_host */, CUDABlasSide(side),
3121 CUDABlasUpperLower(uplo), CUDABlasTranspose(transa),
3122 CUDABlasDiagonal(diag), m, n, &alpha, GpuMemory(a), lda,
3123 GpuMemoryMutable(b), ldb);
3124}
3125
3126bool CUDABlas::DoBlasTrsm(Stream *stream, blas::Side side,
3127 blas::UpperLower uplo, blas::Transpose transa,

Callers

nothing calls this directly

Calls 7

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

Tested by

no test coverage detected