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

Method DoBlasHbmv

tensorflow/stream_executor/cuda/cuda_blas.cc:1193–1204  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1191}
1192
1193bool CUDABlas::DoBlasHbmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1194 uint64 k, std::complex<float> alpha,
1195 const DeviceMemory<std::complex<float>> &a, int lda,
1196 const DeviceMemory<std::complex<float>> &x, int incx,
1197 std::complex<float> beta,
1198 DeviceMemory<std::complex<float>> *y, int incy) {
1199 return DoBlasInternal<std::complex<float>>(
1200 cublasChbmv, stream, true /* = pointer_mode_host */,
1201 CUDABlasUpperLower(uplo), n, k, GpuComplex(&alpha),
1202 GpuComplex(GpuMemory(a)), lda, GpuComplex(GpuMemory(x)), incx,
1203 GpuComplex(&beta), GpuComplex(GpuMemoryMutable(y)), incy);
1204}
1205
1206bool CUDABlas::DoBlasHbmv(Stream *stream, blas::UpperLower uplo, uint64 n,
1207 uint64 k, 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