composable_kernel icon indicating copy to clipboard operation
composable_kernel copied to clipboard

Published 20 hours ago • verified publisher icon ROCmSoftwarePlatform

Refactor ck_tile fMHA forward example

Open poyenc opened this issue 10 months ago • 2 comments

poyenc avatar Apr 17 '24 16:04 poyenc

we need to wait for @danyao12 merge his fmha bwd & dropout changes then refactor all the updated example codes together.

poyenc avatar Apr 25 '24 01:04 poyenc

I will continue developing the fmha fwd + KV cache reference function base on current design of HostTensor<>.

poyenc avatar Apr 25 '24 13:04 poyenc

this PR is no longer needed.

poyenc avatar Sep 20 '24 07:09 poyenc