4 people 5bfd1bbc98 [Kernel] Adding bias epilogue support for cutlass_scaled_mm (#5560) ...

Co-authored-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>
Co-authored-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>

2024-06-26 15:16:00 +00: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] Fix CUTLASS 3.x custom broadcast load epilogue (#5516)

2024-06-14 09:30:15 -07:00

common.hpp

[BugFix] [Kernel] Add Cutlass2x fallback kernels (#5744)

2024-06-23 21:07:11 +00:00

scaled_mm_c2x.cu

[Kernel] Adding bias epilogue support for cutlass_scaled_mm (#5560)

2024-06-26 15:16:00 +00:00

scaled_mm_c3x.cu

[Kernel] Adding bias epilogue support for cutlass_scaled_mm (#5560)

2024-06-26 15:16:00 +00:00

scaled_mm_entry.cu

[Kernel] Adding bias epilogue support for cutlass_scaled_mm (#5560)

2024-06-26 15:16:00 +00:00