Qianfeng e3f44659cf Fix in dropout lambda to avoid the compiling issue on some docker/compiler envs (#1350)

2024-06-20 11:36:42 +08: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

Fix in dropout lambda to avoid the compiling issue on some docker/compiler envs (#1350)

2024-06-20 11:36:42 +08:00

gemm

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

reduce/block

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05: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

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

gemm.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

reduce.hpp

introducing ck_tile! (#1216)

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