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

Function CUDABlasUpperLower

tensorflow/stream_executor/cuda/cuda_blas.cc:320–329  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

318}
319
320cublasFillMode_t CUDABlasUpperLower(blas::UpperLower uplo) {
321 switch (uplo) {
322 case blas::UpperLower::kUpper:
323 return CUBLAS_FILL_MODE_UPPER;
324 case blas::UpperLower::kLower:
325 return CUBLAS_FILL_MODE_LOWER;
326 default:
327 LOG(FATAL) << "Invalid value of blas::UpperLower.";
328 }
329}
330
331cublasDiagType_t CUDABlasDiagonal(blas::Diagonal diag) {
332 switch (diag) {

Callers 15

DoBlasHbmvMethod · 0.70
DoBlasHemvMethod · 0.70
DoBlasHerMethod · 0.70
DoBlasHer2Method · 0.70
DoBlasHpmvMethod · 0.70
DoBlasHprMethod · 0.70
DoBlasHpr2Method · 0.70
DoBlasSbmvMethod · 0.70
DoBlasSpmvMethod · 0.70
DoBlasSprMethod · 0.70
DoBlasSpr2Method · 0.70
DoBlasSymvMethod · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected