WMMA gemm_add_relu_add_layernorm (#2989)

Browse Source

* Summary:

 - Refactor epilogue (with CShuffle) to support fused operations:
    - EpilogueCShuffleBase holds common parts
    - EpilogueCShuffle: runs CShuffle and write out
    - EpilogueWelfordCShuffle: holds Welford specific arguments, runs CShuffle, write out, Welford first part and Welford write out

 - Extend thread transfer v7r3:
    - Support for intermediate data type different from src and dst type
    - New functionality to write to dst buffer and keep data (to be able to use them for additional operations)

* Adress review comments

...

This commit is contained in:

Enrico Degregori

2025-10-31 19:19:26 +01:00

committed by GitHub

parent e9596228ff

commit 4ebc48a3cd

23 changed files with 2678 additions and 332 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

include/ck_tile/host/tensor_shuffle_utils.hpp Executable file → Normal file