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

Function CUDABlasSide

tensorflow/stream_executor/cuda/cuda_blas.cc:342–351  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

340}
341
342cublasSideMode_t CUDABlasSide(blas::Side side) {
343 switch (side) {
344 case blas::Side::kLeft:
345 return CUBLAS_SIDE_LEFT;
346 case blas::Side::kRight:
347 return CUBLAS_SIDE_RIGHT;
348 default:
349 LOG(FATAL) << "Invalid value of blas::Side.";
350 }
351}
352
353// CUDADataType<T>::type translates from a C++ type (e.g. float) to a
354// cudaDataType_t (e.g. CUDA_R_32F). CUDAComputationType(ty) translates from a

Callers 4

DoBlasHemmMethod · 0.85
DoBlasSymmMethod · 0.85
DoBlasTrmmMethod · 0.85
DoBlasTrsmMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected