csrc/quantization/cutlass_w8a8/scaled_mm_entry.cu · a7dcc62086ea751b46b4821c2811cf8ac83711bf · OpenDAS / vllm_cscc

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

Tyler Michael Smith authored Jun 13, 2024


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>

85657b56

scaled_mm_entry.cu 2.94 KB

Replace scaled_mm_entry.cu