Skip to content

GitLab

"host/online_compile/CMakeLists.txt" did not exist on "5c7cec11159d22636dd4c1119e7e430d156a8df7"
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
simplify_reshapes_test.cpp 59.9 KB