Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions hls4ml/templates/vivado/myproject_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@ size_t trace_type_size = sizeof(double);
int main(int argc, char **argv) {
// hls-fpga-machine-learning insert namespace

// hls-fpga-machine-learning insert load weights

// load input data from text file
std::ifstream fin("tb_data/tb_input_features.dat");
// load predictions from text file
Expand Down
10 changes: 10 additions & 0 deletions hls4ml/writer/vivado_writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -220,6 +220,8 @@ def write_project_cpp(self, model):
newline += indent + '#pragma HLS INTERFACE ap_vld port={},{} \n'.format(
','.join(all_inputs), ','.join(all_outputs)
)
if all_brams:
newline += indent + '#pragma HLS INTERFACE bram port={} \n'.format(','.join(all_brams))
newline += pipeline_pragma

if io_type == 'io_stream':
Expand Down Expand Up @@ -586,6 +588,14 @@ def write_test_bench(self, model):
for bram in model_brams:
newline += f'#include "firmware/weights/{bram.name}.h"\n'

elif '// hls-fpga-machine-learning insert load weights' in line:
newline = line
if model_brams and model.config.get_writer_config()['WriteWeightsTxt']:
for bram in model_brams:
newline += indent + 'nnet::load_weights_from_txt<{}, {}>({}, "{}.txt");\n'.format(
bram.type.name, bram.data_length, bram.name, bram.name
)

elif '// hls-fpga-machine-learning insert data' in line:
newline = line
offset = 0
Expand Down
Loading