From 89da402e27d09627329e0d6baffe6f75ad6a1e52 Mon Sep 17 00:00:00 2001 From: Francesco Conti Date: Mon, 19 Feb 2024 12:57:33 +0000 Subject: [PATCH] Make N-EUREKA size parametric (spatial dims) Through the NNX_NEUREKA_PE_{H,W} defines. --- neureka/hal/neureka_task_defs.h | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/neureka/hal/neureka_task_defs.h b/neureka/hal/neureka_task_defs.h index fa08289..dfd74d4 100644 --- a/neureka/hal/neureka_task_defs.h +++ b/neureka/hal/neureka_task_defs.h @@ -21,18 +21,21 @@ #ifndef __NEUREKA_DEFS_H__ #define __NEUREKA_DEFS_H__ -/* ARHITECTURE */ +/* ARCHITECTURE */ -#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (6) -#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (6) +#define NNX_NEUREKA_PE_H (6) +#define NNX_NEUREKA_PE_W (6) + +#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (NNX_NEUREKA_PE_H) +#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (NNX_NEUREKA_PE_W) #define NEUREKA_SUBTILE_INPUT_CHANNEL_1x1 (32) -#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (8) -#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (8) -#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (28) +#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (NNX_NEUREKA_PE_H+2) +#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (NNX_NEUREKA_PE_W+2) +#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (32) -#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (6) -#define NEUREKA_SUBTILE_OUTPUT_WIDTH (6) +#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (4) +#define NEUREKA_SUBTILE_OUTPUT_WIDTH (4) #define NEUREKA_SUBTILE_OUTPUT_CHANNEL (32) #define NEUREKA_OUTPUT_BANDWIDTH_BYTES (32)