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

Method DoBlasSpmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1377–1385  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1375}
1376
1377bool CUDABlas::DoBlasSpmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1378 float alpha, const DeviceMemory<float> &ap,
1379 const DeviceMemory<float> &x, int incx, float beta,
1380 DeviceMemory<float> *y, int incy) {
1381 return DoBlasInternal<float>(
1382 cublasSspmv, stream, true /* = pointer_mode_host */,
1383 CUDABlasUpperLower(uplo), n, &alpha, GpuMemory(ap), GpuMemory(x), incx,
1384 &beta, GpuMemoryMutable(y), incy);
1385}
1386
1387bool CUDABlas::DoBlasSpmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1388 double alpha, const DeviceMemory<double> &ap,

Callers

nothing calls this directly

Calls 3

GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
CUDABlasUpperLowerFunction · 0.70

Tested by

no test coverage detected