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

Method DoBlasAxpy

tensorflow/stream_executor/cuda/cuda_blas.cc:623–629  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

621}
622
623bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha,
624 const DeviceMemory<float> &x, int incx,
625 DeviceMemory<float> *y, int incy) {
626 return DoBlasInternal<float>(
627 cublasSaxpy, stream, true /* = pointer_mode_host */, elem_count, &alpha,
628 GpuMemory(x), incx, GpuMemoryMutable(y), incy);
629}
630
631bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, double alpha,
632 const DeviceMemory<double> &x, int incx,

Callers

nothing calls this directly

Calls 3

GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
GpuComplexFunction · 0.85

Tested by

no test coverage detected