5 people 85657b5607 [Kernel] Factor out epilogues from cutlass kernels (#5391) ...

Co-authored-by: Michael Goin <michael@neuralmagic.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: zifeitong <zifei.tong@parasail.io>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-neuralmagic@users.noreply.github.com>

2024-06-13 11:22:19 -07:00

..

broadcast_load_epilogue_c2x.hpp

[Kernel] Refactor CUTLASS kernels to always take scales that reside on the GPU (#5137)

2024-06-01 06:45:32 +00:00

broadcast_load_epilogue_c3x.hpp

[Kernel] Refactor CUTLASS kernels to always take scales that reside on the GPU (#5137)

2024-06-01 06:45:32 +00:00

common.hpp

[Kernel] Add w8a8 CUTLASS kernels (#4749)

2024-05-16 18:32:50 -04:00

scaled_mm_c2x.cu

[Kernel] Factor out epilogues from cutlass kernels (#5391)

2024-06-13 11:22:19 -07:00

scaled_mm_c3x.cu

[Kernel] Factor out epilogues from cutlass kernels (#5391)

2024-06-13 11:22:19 -07:00

scaled_mm_entry.cu

[Kernel] Factor out epilogues from cutlass kernels (#5391)

2024-06-13 11:22:19 -07:00