Skip to content

GitLab

Commit b82522da authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 0b639e94
Hide whitespace changes
Inline Side-by-side
Showing with 21 additions and 20 deletions
src/targets/gpu/gemm.cpp
View file @ b82522da
...@@ -118,27 +118,28 @@ argument miopen_gemm::compute(context& ctx, ...@@ -118,27 +118,28 @@ argument miopen_gemm::compute(context& ctx,
auto beta_r = to_rocblas_type(as(beta)); auto beta_r = to_rocblas_type(as(beta));
auto to_pointer = [&](auto&& arg) { return to_rocblas_type(as.from(arg.data())); }; auto to_pointer = [&](auto&& arg) { return to_rocblas_type(as.from(arg.data())); };
// call the strided implementation only if there are multiple matrices // call the strided implementation only if there are multiple matrices
if (batch_num > 1) if(batch_num > 1)
{ {
generic_rocblas_batched_gemm(as, generic_rocblas_batched_gemm(
ctx.get_stream().get_rocblas(), as,
transb ? rocblas_operation_transpose : rocblas_operation_none, ctx.get_stream().get_rocblas(),
transa ? rocblas_operation_transpose : rocblas_operation_none, transb ? rocblas_operation_transpose : rocblas_operation_none,
n, transa ? rocblas_operation_transpose : rocblas_operation_none,
m, n,
k, m,
&alpha_r, k,
to_pointer(args[1]), &alpha_r,
ldb, to_pointer(args[1]),
k * n, ldb,
to_pointer(args[0]), k * n,
lda, to_pointer(args[0]),
m * k, lda,
&beta_r, m * k,
to_pointer(args[2]), &beta_r,
ldc, to_pointer(args[2]),
m * n, ldc,
batch_num); m * n,
batch_num);
} }
else else
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment