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

Method DoBlasGemmWithAlgorithm

tensorflow/stream_executor/cuda/cuda_blas.cc:2244–2254  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2242}
2243
2244bool CUDABlas::DoBlasGemmWithAlgorithm(
2245 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,
2246 uint64 n, uint64 k, const HostOrDeviceScalar<int> &alpha,
2247 const DeviceMemory<int8> &a, int lda, const DeviceMemory<int8> &b, int ldb,
2248 const HostOrDeviceScalar<int> &beta, DeviceMemory<int> *c, int ldc,
2249 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
2250 blas::ProfileResult *output_profile_result) {
2251 return DoBlasGemmWithAlgorithmImpl(
2252 stream, transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc,
2253 computation_type, algorithm, output_profile_result);
2254}
2255
2256bool CUDABlas::DoBlasGemmWithAlgorithm(
2257 Stream *stream, blas::Transpose transa, blas::Transpose transb, uint64 m,

Callers

nothing calls this directly

Calls 2

is_pointerMethod · 0.45
valueMethod · 0.45

Tested by

no test coverage detected