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

Method DoBlasHer2

tensorflow/stream_executor/cuda/cuda_blas.cc:1265–1275  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1263}
1264
1265bool CUDABlas::DoBlasHer2(Stream *stream, blas::UpperLower uplo, uint64 n,
1266 std::complex<float> alpha,
1267 const DeviceMemory<std::complex<float>> &x, int incx,
1268 const DeviceMemory<std::complex<float>> &y, int incy,
1269 DeviceMemory<std::complex<float>> *a, int lda) {
1270 return DoBlasInternal<std::complex<float>>(
1271 cublasCher2, stream, true /* = pointer_mode_host */,
1272 CUDABlasUpperLower(uplo), n, GpuComplex(&alpha), GpuComplex(GpuMemory(x)),
1273 incx, GpuComplex(GpuMemory(y)), incy, GpuComplex(GpuMemoryMutable(a)),
1274 lda);
1275}
1276
1277bool CUDABlas::DoBlasHer2(Stream *stream, blas::UpperLower uplo, uint64 n,
1278 std::complex<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