diff --git a/example/ck_tile/42_unified_attention/README.md b/example/ck_tile/42_unified_attention/README.md index 438ae1f9ce..aa8a996889 100644 --- a/example/ck_tile/42_unified_attention/README.md +++ b/example/ck_tile/42_unified_attention/README.md @@ -16,6 +16,7 @@ This will result in an executable `build/bin/tile_example_unified_attention` The kernel template is `unified_attention.hpp`, this is the grid-wise op in old ck_tile's terminology. We put it here purposely, to demonstrate one can construct a kernel by using various internal component from ck_tile. We may still have an implementation under ck_tile's include path (in the future) for the kernel template. There are 2 template parameters for this kernel template. + * `FmhaPipeline` is one of the block_tile_pipeline(under `include/ck_tile/tile_program/block_tile_pipeline`) which is a performance critical component. Indeed, we did a lot of optimization and trials to optimize the pipeline and may still workout more performance pipeline and update into that folder. People only need to replace this pipeline type and would be able to enjoy the benefit of different performant implementations (stay tuned for updated pipeline(s)). * `EpiloguePipeline` is the last stage of the pipeline. It modifies and stores the result. Post-fusion can be done at this stage though the example only returns the result.