vllm/csrc/quantization/cutlass_w8a8
2024-10-03 22:55:25 -04:00
..
broadcast_load_epilogue_c2x.hpp [Kernel] Add per-tensor and per-token AZP epilogues (#5941) 2024-08-06 18:17:08 +00:00
broadcast_load_epilogue_c3x.hpp [Kernel] Update CUTLASS to 3.5.1 (#7085) 2024-08-05 15:13:43 -04:00
common.hpp [BugFix] [Kernel] Add Cutlass2x fallback kernels (#5744) 2024-06-23 21:07:11 +00:00
Epilogues.md [Kernel] Add per-tensor and per-token AZP epilogues (#5941) 2024-08-06 18:17:08 +00:00
scaled_mm_c2x_sm75_dispatch.cuh [Kernel] Tuned int8 Cutlass Kernels for SM75 (T4) (#6996) 2024-07-31 14:40:32 -07:00
scaled_mm_c2x_sm80_dispatch.cuh [Kernel] Tuned FP8 Kernels for Ada Lovelace (#6677) 2024-07-29 09:42:35 -06:00
scaled_mm_c2x_sm89_fp8_dispatch.cuh [Kernel] Tuned int8 kernels for Ada Lovelace (#6848) 2024-07-29 20:24:58 -06:00
scaled_mm_c2x_sm89_int8_dispatch.cuh [Kernel] Tuned int8 kernels for Ada Lovelace (#6848) 2024-07-29 20:24:58 -06:00
scaled_mm_c2x.cu [Kernel] Add per-tensor and per-token AZP epilogues (#5941) 2024-08-06 18:17:08 +00:00
scaled_mm_c2x.cuh [Kernel] Add per-tensor and per-token AZP epilogues (#5941) 2024-08-06 18:17:08 +00:00
scaled_mm_c3x.cu [Kernel] Add per-tensor and per-token AZP epilogues (#5941) 2024-08-06 18:17:08 +00:00
scaled_mm_entry.cu [CI/Build] Per file CUDA Archs (improve wheel size and dev build times) (#8845) 2024-10-03 22:55:25 -04:00