vllm/csrc/quantization/cutlass_w8a8
Luka Govedič 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