schihei/mixed-precision-neural-networks-on-risc-v-cores
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix generate_opt_c_code_mlp

Browse Source
This commit is contained in:
Alexios Maras
2024-07-26 20:16:18 +03:00
committed by GitHub
parent bfb12723e3
commit 487d3bd3dc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mpq/configure_ibex.py
+1 -1
View File
@@ -866,7 +866,7 @@ def generate_opt_c_code_mlp(path, name, int_weights, optimal_config, type_of_lay
f.write('#include "ibex_inputs.h"\n\n') f.write('#include "ibex_inputs.h"\n\n')
f.write('#define IN_DIM ' + str((8//optimal_config[0]) * int_weights[0].shape[1])) f.write('#define IN_DIM ' + str((8//optimal_config[0]) * int_weights[0].shape[1]))
for i in range(1, len(int_weights)): for i in range(1, len(int_weights)):
f.write('\n#define HIDDEN_DIM' + str(i) + ' ' + str(4 * int_weights[i].shape[1])) f.write('\n#define HIDDEN_DIM' + str(i) + ' ' + str(4 * int_weights[i-1].shape[0]))
f.write('\n#define OUT_DIM ' + str(4 * int_weights[-1].shape[0])) f.write('\n#define OUT_DIM ' + str(4 * int_weights[-1].shape[0]))
f.write('\n#define SAMPLES 1\n\n') f.write('\n#define SAMPLES 1\n\n')