cutlass/examples/42_fused_multi_head_attention at 3c90f6aea6a93841fc67eb147a7d6602e6b7c28e - cutlass - Gitea: Git with a cup of tea

leavelet/cutlass

Files

T

History

Wenzhuo Liu 3c90f6aea6 add #pragma once for header file in example 42 (#698 )

2022-11-15 22:50:24 -05:00

..

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

attention_scaling_coefs_updater.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

CMakeLists.txt

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

debug_utils.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

epilogue_pipelined.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

epilogue_rescale_output.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

epilogue_thread_apply_logsumexp.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

find_default_mma.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

fused_multihead_attention.cu

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

gemm_kernel_utils.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00

kernel_forward.h

add #pragma once for header file in example 42 (#698 )

2022-11-15 22:50:24 -05:00

mma_from_smem.h

ex42: Fused MHA imported from xFormers (#662 )

2022-10-17 10:49:33 -04:00