Skip to content

GitLab

"sgl-kernel/vscode:/vscode.git/clone" did not exist on "bc20e93f2df15f81e624cd624e693406180e9178"
Switch branch/tag
Find file BlameHistoryPermalink
  • turneram's avatar
    Use rocblas_gemm_ex for batched gemms with broadcasted B (#1354) · a10a8ef1
    turneram authored 
    Improves performance for 4/6 GEMMs used by huggingface BERT models with batch_size>1 by using a non-batched rocBLAS call for GEMMs where the B input has a broadcasted batch dimension.
The four verify tests added reflect the actual configurations used by bert-base-cased, with varied batch sizes.

Also adds a matcher to simplify_reshapes to move multibroadcasts after concats.
    a10a8ef1
gemm_impl.cpp 10.1 KB