Skip to content

Commit f0f8967

Browse files
author
georg
committed
wrap Neureka architecture parameters in #ifndefs
1 parent 7bec621 commit f0f8967

File tree

1 file changed

+16
-7
lines changed

1 file changed

+16
-7
lines changed

neureka/hal/neureka_task_defs.h

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -21,20 +21,29 @@
2121
#ifndef __NEUREKA_DEFS_H__
2222
#define __NEUREKA_DEFS_H__
2323

24-
/* ARHITECTURE */
25-
26-
#define NNX_NEUREKA_PE_H (4)
27-
#define NNX_NEUREKA_PE_W (4)
24+
/* ARCHITECTURE */
25+
// The definitions wrapped in #ifndefs can be overwritten with compiler flags for
26+
// different parametrizations of the Neureka architecture
27+
#ifndef NNX_NEUREKA_PE_H
28+
#define NNX_NEUREKA_PE_H (6)
29+
#endif
30+
#ifndef NNX_NEUREKA_PE_W
31+
#define NNX_NEUREKA_PE_W (6)
32+
#endif
33+
#ifndef NNX_NEUREKA_BANDWIDTH_1x1
2834
#define NNX_NEUREKA_BANDWIDTH_1x1 (256)
29-
#define NNX_NEUREKA_BANDWIDTH_3x3 (288)
35+
#endif
36+
#define NNX_NEUREKA_BANDWIDTH_3x3 (256)
3037

3138
#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (NNX_NEUREKA_PE_H)
3239
#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (NNX_NEUREKA_PE_W)
3340
#define NEUREKA_SUBTILE_INPUT_CHANNEL_1x1 (32)
3441

3542
#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (NNX_NEUREKA_PE_H+2)
36-
#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (NNX_NEUREKA_PE_W+2)
37-
#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (32)
43+
#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (NNX_NEUREKA_PE_W + 2)
44+
#ifndef NEUREKA_SUBTILE_INPUT_CHANNEL_3x3
45+
#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (28)
46+
#endif
3847

3948
#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (NNX_NEUREKA_PE_H)
4049
#define NEUREKA_SUBTILE_OUTPUT_WIDTH (NNX_NEUREKA_PE_W)

0 commit comments

Comments
 (0)