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

Method DoBlasHpr

tensorflow/stream_executor/cuda/cuda_blas.cc:1315–1323  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1313}
1314
1315bool CUDABlas::DoBlasHpr(Stream *stream, blas::UpperLower uplo, uint64 n,
1316 float alpha,
1317 const DeviceMemory<std::complex<float>> &x, int incx,
1318 DeviceMemory<std::complex<float>> *ap) {
1319 return DoBlasInternal<std::complex<float>>(
1320 cublasChpr, stream, true /* = pointer_mode_host */,
1321 CUDABlasUpperLower(uplo), n, GpuComplex(&alpha), GpuComplex(GpuMemory(x)),
1322 incx, GpuComplex(GpuMemoryMutable(ap)));
1323}
1324
1325bool CUDABlas::DoBlasHpr(Stream *stream, blas::UpperLower uplo, uint64 n,
1326 double alpha,

Callers

nothing calls this directly

Calls 4

GpuComplexFunction · 0.85
GpuMemoryFunction · 0.85
GpuMemoryMutableFunction · 0.85
CUDABlasUpperLowerFunction · 0.70

Tested by

no test coverage detected