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

Method DoBlasHemm

tensorflow/stream_executor/cuda/cuda_blas.cc:2830–2842  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2828}
2829
2830bool CUDABlas::DoBlasHemm(Stream *stream, blas::Side side,
2831 blas::UpperLower uplo, uint64 m, uint64 n,
2832 std::complex<float> alpha,
2833 const DeviceMemory<std::complex<float>> &a, int lda,
2834 const DeviceMemory<std::complex<float>> &b, int ldb,
2835 std::complex<float> beta,
2836 DeviceMemory<std::complex<float>> *c, int ldc) {
2837 return DoBlasInternal<std::complex<float>>(
2838 cublasChemm, stream, true /* = pointer_mode_host */, CUDABlasSide(side),
2839 CUDABlasUpperLower(uplo), m, n, GpuComplex(&alpha),
2840 GpuComplex(GpuMemory(a)), lda, GpuComplex(GpuMemory(b)), ldb,
2841 GpuComplex(&beta), GpuComplex(GpuMemoryMutable(c)), ldc);
2842}
2843
2844bool CUDABlas::DoBlasHemm(Stream *stream, blas::Side side,
2845 blas::UpperLower uplo, uint64 m, uint64 n,

Callers

nothing calls this directly

Calls 5

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

Tested by

no test coverage detected