PoYen, Chen 381f7e90e0 Merge branch 'develop' into feature/fmha-fwd-appendkv

2024-08-04 02:12:20 +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

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

2024-08-04 02:12:20 +00:00

gemm

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05: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

Rename header

2024-07-23 07:45:25 +00:00

gemm.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05: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