From 0b86a6345e3ec3014a2f96a5fcd35052ee56aeea Mon Sep 17 00:00:00 2001 From: Cautiousss <48676630+Cautiousss@users.noreply.github.com> Date: Fri, 13 May 2022 22:00:05 +0800 Subject: [PATCH] [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu code style (#953) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: 何晓昕 --- colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu b/colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu index b12437870..58d26235a 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu +++ b/colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu @@ -16,7 +16,7 @@ __global__ void ls_cross_entropy_fw_kernel( const int left_idx = block_start + threadIdx.x; const int right_idx = (blockIdx.x + 1) * vocab_size; float max_input[1] = {REDUCE_FLOAT_INF_NEG}; - float sum_logits[2] = {0.f, 0.f}; // logit and logit exp + float sum_logits[2] = {0.f, 0.f}; // logit and logit exp int target_tid = targets[blockIdx.x]; if (target_tid == padding_idx) {