Skip to content

Commit 794e371

Browse files
committed
Fix concatenate1d for io_stream
1 parent 631878c commit 794e371

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

hls4ml/templates/vivado/nnet_utils/nnet_merge_stream.h

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -329,9 +329,9 @@ void concatenate2d_0(
329329
ConcatPackInput2: for (int k = 0; k < input2_T::size; k++) {
330330
#pragma HLS UNROLL
331331
out_data[k] = in_data2[k];
332-
}
332+
}
333333

334-
res.write(out_data);
334+
res.write(out_data);
335335
}
336336
}
337337

@@ -352,14 +352,14 @@ void concatenate2d_1(
352352
ConcatPackInput1: for (int k = 0; k < input1_T::size; k++) {
353353
#pragma HLS UNROLL
354354
out_data[k] = in_data1[k];
355-
}
355+
}
356356

357-
ConcatPackInput2: for (int k = 0; k < input2_T::size; k++) {
357+
ConcatPackInput2: for (int k = 0; k < input2_T::size; k++) {
358358
#pragma HLS UNROLL
359359
out_data[input1_T::size + k] = in_data2[k];
360-
}
360+
}
361361

362-
res.write(out_data);
362+
res.write(out_data);
363363
}
364364
}
365365

@@ -385,23 +385,23 @@ void concatenate1d(
385385
res_T out_data;
386386
#pragma HLS DATA_PACK variable=out_data
387387
ConcatLoop1: for (int i = 0; i < CONFIG_T::n_elem1_0 / input1_T::size; i++) {
388-
#pragma HLS PIPELINE
388+
#pragma HLS PIPELINE
389389
input1_T in_data1 = data1.read();
390-
ConcatPack1: for (int j = 0; j < res_T::size; j++) {
390+
ConcatPack1: for (int j = 0; j < input1_T::size; j++) {
391391
#pragma HLS UNROLL
392-
out_data[j] = in_data1[j];
392+
out_data[j] = in_data1[j];
393393
}
394-
res.write(out_data);
395394
}
396395
ConcatLoop2: for (int i = 0; i < CONFIG_T::n_elem2_0 / input2_T::size; i++) {
397-
#pragma HLS PIPELINE
396+
#pragma HLS PIPELINE
398397
input2_T in_data2 = data2.read();
399-
ConcatPack2: for (int j = 0; j < res_T::size; j++) {
398+
ConcatPack2: for (int j = 0; j < input2_T::size; j++) {
400399
#pragma HLS UNROLL
401-
out_data[j] = in_data2[j];
400+
out_data[input1_T::size + j] = in_data2[j];
402401
}
403-
res.write(out_data);
402+
404403
}
404+
res.write(out_data);
405405
}
406406
}
407407

0 commit comments

Comments
 (0)