47 lines
1.9 KiB
C
47 lines
1.9 KiB
C
/* SPDX-License-Identifier: AGPL-3.0-or-later */
|
|
/* Copyright (C) 2026 David (Corey) Herrera */
|
|
/**
|
|
* @file det_math_arm.c
|
|
* @brief Instantiates kernels/ft_det_math.h as a DetMathArm table. Compiled once per arm:
|
|
* no flags -> det_math_arm_production(); with DET_MATH_ARM_FN / DET_MATH_ARM_NAME and
|
|
* an FT_DET_* constant override -> a red arm (see Makefile.detmath). The same file is
|
|
* the rv32 text-size probe (`make -f Makefile.detmath size`).
|
|
*
|
|
* @status None
|
|
* @issues None
|
|
* @todo None
|
|
*/
|
|
#include "../kernels/ft_det_math.h"
|
|
#include "det_math_arm.h"
|
|
|
|
#ifndef DET_MATH_ARM_FN
|
|
#define DET_MATH_ARM_FN det_math_arm_production
|
|
#define DET_MATH_ARM_NAME "production"
|
|
#endif
|
|
|
|
static uint32_t arm_add(uint32_t a, uint32_t b) { return ft_det_add(a, b); }
|
|
static uint32_t arm_mul(uint32_t a, uint32_t b) { return ft_det_mul(a, b); }
|
|
static uint32_t arm_exp(uint32_t x) { return ft_det_exp(x); }
|
|
static uint32_t arm_softplus(uint32_t x) { return ft_det_softplus(x); }
|
|
static uint32_t arm_sigmoid(uint32_t x) { return ft_det_sigmoid(x); }
|
|
static uint32_t arm_gdn_beta(uint32_t b) { return ft_det_gdn_beta(b); }
|
|
static uint32_t arm_gdn_g(uint32_t alpha, uint32_t dt, uint32_t ssm_a)
|
|
{
|
|
return ft_det_gdn_g(alpha, dt, ssm_a);
|
|
}
|
|
static uint32_t arm_gdn_decay(uint32_t g) { return ft_det_gdn_decay(g); }
|
|
static uint32_t arm_gdn_beta_reference(uint32_t b) { return ft_det_gdn_beta_reference(b); }
|
|
static uint32_t arm_gdn_gate(uint32_t alpha, uint32_t dt, uint32_t ssm_a)
|
|
{
|
|
return ft_det_gdn_gate(alpha, dt, ssm_a);
|
|
}
|
|
|
|
const DetMathArm *DET_MATH_ARM_FN(void)
|
|
{
|
|
static const DetMathArm arm = {DET_MATH_ARM_NAME, arm_add, arm_mul,
|
|
arm_exp, arm_softplus, arm_sigmoid,
|
|
arm_gdn_beta, arm_gdn_beta_reference,
|
|
arm_gdn_g, arm_gdn_decay, arm_gdn_gate};
|
|
return &arm;
|
|
}
|