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

Function CUDABlasTranspose

tensorflow/stream_executor/cuda/cuda_blas.cc:307–318  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

305// Helper functions transforming blas arguments into cuBLAS arguments.
306
307cublasOperation_t CUDABlasTranspose(blas::Transpose trans) {
308 switch (trans) {
309 case blas::Transpose::kNoTranspose:
310 return CUBLAS_OP_N;
311 case blas::Transpose::kTranspose:
312 return CUBLAS_OP_T;
313 case blas::Transpose::kConjugateTranspose:
314 return CUBLAS_OP_C;
315 default:
316 LOG(FATAL) << "Invalid value of blas::Transpose.";
317 }
318}
319
320cublasFillMode_t CUDABlasUpperLower(blas::UpperLower uplo) {
321 switch (uplo) {

Callers 15

DoBlasGbmvMethod · 0.85
DoBlasGemvMethod · 0.85
DoBlasTbmvMethod · 0.85
DoBlasTbsvMethod · 0.85
DoBlasTpmvMethod · 0.85
DoBlasTpsvMethod · 0.85
DoBlasTrmvMethod · 0.85
DoBlasTrsvMethod · 0.85
DoBlasGemmMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected