 Tadej CiglaričandAli Hassani
|
6f55278121
|
bugfix generic-k code in top-k with softmax (#1993)
* bugfix generic-k code in top-k with softmax
* Update include/cutlass/epilogue/fusion/sm90_visitor_topk_softmax.hpp
Co-authored-by: Ali Hassani <68103095+alihassanijr@users.noreply.github.com>
* Update examples/61_hopper_gemm_with_topk_and_softmax/61_hopper_gemm_with_topk_and_softmax.cu
Co-authored-by: Ali Hassani <68103095+alihassanijr@users.noreply.github.com>
---------
Co-authored-by: Ali Hassani <68103095+alihassanijr@users.noreply.github.com>
|
2025-01-31 19:05:35 -05:00 |
|