squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
922fb5108b
cutlass/examples/41_fused_multi_head_attention/epilogue
History
dan_the_3rd f303889ed9
fMHA: Sync FW with xFormers (#828) 
* fMHA: Add support for bias+dropout in FW

* Remove 'getMaximumSharedMemoryPerBlockKb'

* fix comments

---------

Co-authored-by: danthe3rd <danthe3rd>
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2023-02-22 23:25:31 -05:00
..
epilogue_pipelined.h fMHA: Sync FW with xFormers (#828) 2023-02-22 23:25:31 -05:00
epilogue_rescale_output.h fMHA: Sync FW with xFormers (#828) 2023-02-22 23:25:31 -05:00
epilogue_thread_apply_logsumexp.h fMHA: Sync FW with xFormers (#828) 2023-02-22 23:25:31 -05:00