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

Method DoBlasSymm

tensorflow/stream_executor/cuda/cuda_blas.cc:2914–2923  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2912}
2913
2914bool CUDABlas::DoBlasSymm(Stream *stream, blas::Side side,
2915 blas::UpperLower uplo, uint64 m, uint64 n,
2916 float alpha, const DeviceMemory<float> &a, int lda,
2917 const DeviceMemory<float> &b, int ldb, float beta,
2918 DeviceMemory<float> *c, int ldc) {
2919 return DoBlasInternal<float>(
2920 cublasSsymm, stream, true /* = pointer_mode_host */, CUDABlasSide(side),
2921 CUDABlasUpperLower(uplo), m, n, &alpha, GpuMemory(a), lda, GpuMemory(b),
2922 ldb, &beta, GpuMemoryMutable(c), ldc);
2923}
2924
2925bool CUDABlas::DoBlasSymm(Stream *stream, blas::Side side,
2926 blas::UpperLower uplo, uint64 m, uint64 n,

Callers

nothing calls this directly

Calls 5

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

Tested by

no test coverage detected