PoYen, Chen d3f550f30c Add s_barrier to sync threads

2024-08-22 09:03:01 +00:00

..

common

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

epilogue

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

fmha

Add s_barrier to sync threads

2024-08-22 09:03:01 +00:00

gemm

[CK_TILE] FA bwd kernels optimization (#1397)

2024-08-16 13:40:10 -07:00

layernorm2d

layernorm2d forward (#1339)

2024-06-24 08:45:52 +08:00

reduce/block

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

welford

layernorm2d forward (#1339)

2024-06-24 08:45:52 +08:00

common.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

epilogue.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

fmha.hpp

Merge branch 'develop' into feature/fmha-fwd-appendkv

2024-08-20 05:49:01 +00:00

gemm.hpp

[CK_TILE] FA bwd kernels optimization (#1397)

2024-08-16 13:40:10 -07:00

layernorm2d.hpp

layernorm2d forward (#1339)

2024-06-24 08:45:52 +08:00

reduce.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

welford.hpp

layernorm2d forward (#1339)

2024-06-24 08:45:52 +08:00