File tree Expand file tree Collapse file tree 1 file changed +16
-7
lines changed
Expand file tree Collapse file tree 1 file changed +16
-7
lines changed Original file line number Diff line number Diff line change 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)
You can’t perform that action at this time.
0 commit comments