Skip to content

Commit 89da402

Browse files
Make N-EUREKA size parametric (spatial dims)
Through the NNX_NEUREKA_PE_{H,W} defines.
1 parent 51868e8 commit 89da402

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

neureka/hal/neureka_task_defs.h

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -21,18 +21,21 @@
2121
#ifndef __NEUREKA_DEFS_H__
2222
#define __NEUREKA_DEFS_H__
2323

24-
/* ARHITECTURE */
24+
/* ARCHITECTURE */
2525

26-
#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (6)
27-
#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (6)
26+
#define NNX_NEUREKA_PE_H (6)
27+
#define NNX_NEUREKA_PE_W (6)
28+
29+
#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (NNX_NEUREKA_PE_H)
30+
#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (NNX_NEUREKA_PE_W)
2831
#define NEUREKA_SUBTILE_INPUT_CHANNEL_1x1 (32)
2932

30-
#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (8)
31-
#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (8)
32-
#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (28)
33+
#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (NNX_NEUREKA_PE_H+2)
34+
#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (NNX_NEUREKA_PE_W+2)
35+
#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (32)
3336

34-
#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (6)
35-
#define NEUREKA_SUBTILE_OUTPUT_WIDTH (6)
37+
#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (4)
38+
#define NEUREKA_SUBTILE_OUTPUT_WIDTH (4)
3639
#define NEUREKA_SUBTILE_OUTPUT_CHANNEL (32)
3740

3841
#define NEUREKA_OUTPUT_BANDWIDTH_BYTES (32)

0 commit comments

Comments
 (0)