diff --git a/test/app/src/nnx_layer.c b/test/app/src/nnx_layer.c index 921c679..696ee31 100644 --- a/test/app/src/nnx_layer.c +++ b/test/app/src/nnx_layer.c @@ -104,18 +104,18 @@ typedef neureka_siracusa_conf_t nnx_bsp_conf_t; #include "weight.h" static void task_prepare(nnx_task_t *task) { - nnx_task_init(task, WEIGHT_HEIGHT, GROUPS > 1, INPUT_BITS, OUTPUT_BITS, - WEIGHT_BITS, weightOffsetModeLayerWise, WEIGHT_OFFSET, - (neureka_quant_t){.shift_amount = OUTSHIFT, - .mode = quantMode8Bit, - .function = HAS_RELU ? quantFunctionRelu - : quantFunctionIdentity, - .flag_rounding = nnxTaskFlagFalse}, - (neureka_norm_t){.mode = normMode8Bit, - .flag_bias = HAS_BIAS ? nnxTaskFlagTrue - : nnxTaskFlagFalse, - .flag_shift = nnxTaskFlagFalse}, - STRIDE_HEIGHT); + nnx_task_init( + task, WEIGHT_HEIGHT, GROUPS > 1, INPUT_BITS, OUTPUT_BITS, WEIGHT_BITS, + weightOffsetModeLayerWise, WEIGHT_OFFSET, + (nnx_quant_t){.shift_amount = OUTSHIFT, + .mode = quantMode8Bit, + .function = + HAS_RELU ? quantFunctionRelu : quantFunctionIdentity, + .flag_rounding = nnxTaskFlagFalse}, + (nnx_norm_t){.mode = normMode8Bit, + .flag_bias = HAS_BIAS ? nnxTaskFlagTrue : nnxTaskFlagFalse, + .flag_shift = nnxTaskFlagFalse}, + STRIDE_HEIGHT); if (STRIDE_WIDTH == 2 && STRIDE_HEIGHT == 2) { nnx_task_set_dims_stride2x2( @@ -147,8 +147,7 @@ static void task_prepare(nnx_task_t *task) { static void task_execute(nnx_task_t *task) { nnx_dev_t *dev = nnx_bsp_get_dev(); - nnx_gvsoc_log_activate(dev, NNX_GVSOC_LOG_LEVEL, - NNX_GVSOC_LOG_FORMAT); + nnx_gvsoc_log_activate(dev, NNX_GVSOC_LOG_LEVEL, NNX_GVSOC_LOG_FORMAT); nnx_bsp_conf_t conf = {.max_stall = 8}; nnx_init(dev, &conf);