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

Method DoBlasSpr2

tensorflow/stream_executor/cuda/cuda_blas.cc:1415–1423  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1413}
1414
1415bool CUDABlas::DoBlasSpr2(Stream *stream, blas::UpperLower uplo, uint64 n,
1416 float alpha, const DeviceMemory<float> &x, int incx,
1417 const DeviceMemory<float> &y, int incy,
1418 DeviceMemory<float> *ap) {
1419 return DoBlasInternal<float>(
1420 cublasSspr2, stream, true /* = pointer_mode_host */,
1421 CUDABlasUpperLower(uplo), n, &alpha, GpuMemory(x), incx, GpuMemory(y),
1422 incy, GpuMemoryMutable(ap));
1423}
1424
1425bool CUDABlas::DoBlasSpr2(Stream *stream, blas::UpperLower uplo, uint64 n,
1426 double alpha, const DeviceMemory<double> &x, int incx,

Callers

nothing calls this directly

Calls 3

GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
CUDABlasUpperLowerFunction · 0.70

Tested by

no test coverage detected