logit_kernels.c File Reference

#include "ckernel_engine.h"
#include <stddef.h>

Go to the source code of this file.

Functions
void	final_logit_scale_f32 (float *logits, int tokens, int vocab_size, float scale)

Function Documentation

final_logit_scale_f32()

void final_logit_scale_f32	(	float *	logits,
		int	tokens,
		int	vocab_size,
		float	scale
	)

Definition at line 7 of file logit_kernels.c.

{
    if (!logits || tokens <= 0 || vocab_size <= 0) {
        return;
    }
 
    const size_t total = (size_t) tokens * (size_t) vocab_size;
    for (size_t i = 0; i < total; ++i) {
        logits[i] *= scale;
    }
}

References vocab_size.