From 70bb71bfd171175129465fbdc7f9438d3a0715b2 Mon Sep 17 00:00:00 2001 From: Sage Moore Date: Thu, 19 Dec 2024 16:43:01 +0000 Subject: [PATCH] fix build Signed-off-by: Sage Moore --- csrc/quantization/activation_kernels.cu | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/csrc/quantization/activation_kernels.cu b/csrc/quantization/activation_kernels.cu index bac41e65e7b30..33610a898af61 100644 --- a/csrc/quantization/activation_kernels.cu +++ b/csrc/quantization/activation_kernels.cu @@ -107,10 +107,9 @@ __global__ void act_and_mul_quant_kernel( scale.data_ptr(), d); \ }); -void silu_and_mul_quant(torch::Tensor& out, // [..., d] - torch::Tensor& input, // [..., 2 * d] - torch::Tensor& scale) -{ +void silu_and_mul_quant(torch::Tensor& out, // [..., d] + torch::Tensor& input, // [..., 2 * d] + torch::Tensor& scale) { TORCH_CHECK(out.dtype() == torch::kFloat8_e4m3fn); TORCH_CHECK(input.dtype() == torch::kFloat16 || input.dtype() == torch::kBFloat16);