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

Method DoBlasSbmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1357–1365  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1355}
1356
1357bool CUDABlas::DoBlasSbmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1358 uint64 k, float alpha, const DeviceMemory<float> &a,
1359 int lda, const DeviceMemory<float> &x, int incx,
1360 float beta, DeviceMemory<float> *y, int incy) {
1361 return DoBlasInternal<float>(
1362 cublasSsbmv, stream, true /* = pointer_mode_host */,
1363 CUDABlasUpperLower(uplo), n, k, &alpha, GpuMemory(a), lda, GpuMemory(x),
1364 incx, &beta, GpuMemoryMutable(y), incy);
1365}
1366
1367bool CUDABlas::DoBlasSbmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1368 uint64 k, double alpha, const DeviceMemory<double> &a,

Callers

nothing calls this directly

Calls 3

GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
CUDABlasUpperLowerFunction · 0.70

Tested by

no test coverage detected