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

Method DoBlasGbmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1035–1044  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1033}
1034
1035bool CUDABlas::DoBlasGbmv(Stream *stream, blas::Transpose trans, uint64 m,
1036 uint64 n, uint64 kl, uint64 ku, float alpha,
1037 const DeviceMemory<float> &a, int lda,
1038 const DeviceMemory<float> &x, int incx, float beta,
1039 DeviceMemory<float> *y, int incy) {
1040 return DoBlasInternal<float>(
1041 cublasSgbmv, stream, true /* = pointer_mode_host */,
1042 CUDABlasTranspose(trans), m, n, kl, ku, &alpha, GpuMemory(a), lda,
1043 GpuMemory(x), incx, &beta, GpuMemoryMutable(y), incy);
1044}
1045
1046bool CUDABlas::DoBlasGbmv(Stream *stream, blas::Transpose trans, uint64 m,
1047 uint64 n, uint64 kl, uint64 ku, double alpha,

Callers

nothing calls this directly

Calls 4

CUDABlasTransposeFunction · 0.85
GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
GpuComplexFunction · 0.85

Tested by

no test coverage detected