Skip to content

Commit c81f736

Browse files
committed
Pipeline padding function
1 parent a3cb4de commit c81f736

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

hls4ml/templates/vivado/nnet_utils/nnet_padding.h

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ void zeropad1d_cf(
1818
data_T data[CONFIG_T::n_chan * CONFIG_T::in_width],
1919
data_T res[CONFIG_T::n_chan * CONFIG_T::out_width]
2020
) {
21+
#pragma HLS PIPELINE
22+
2123
for(int j = 0; j < CONFIG_T::n_chan; j++) {
2224
for (int i = 0; i < CONFIG_T::pad_left; i++) {
2325
*(res++) = 0;
@@ -38,6 +40,8 @@ void zeropad1d_cl(
3840
data_T data[CONFIG_T::n_chan * CONFIG_T::in_width],
3941
res_T res[CONFIG_T::n_chan * CONFIG_T::out_width]
4042
) {
43+
#pragma HLS PIPELINE
44+
4145
for (int i = 0; i < CONFIG_T::pad_left; i++) {
4246
for(int j = 0; j < CONFIG_T::n_chan; j++) {
4347
*(res++) = 0;
@@ -75,6 +79,8 @@ void zeropad2d_cf(
7579
data_T data[CONFIG_T::n_chan * CONFIG_T::in_height * CONFIG_T::in_width],
7680
data_T res[CONFIG_T::n_chan * CONFIG_T::out_height * CONFIG_T::out_width]
7781
) {
82+
#pragma HLS PIPELINE
83+
7884
for(int k = 0; k < CONFIG_T::n_chan; k++) {
7985

8086
for (int i = 0; i < CONFIG_T::pad_top; i++) {
@@ -108,6 +114,8 @@ void zeropad2d_cl(
108114
data_T data[CONFIG_T::n_chan * CONFIG_T::in_height * CONFIG_T::in_width],
109115
res_T res[CONFIG_T::n_chan * CONFIG_T::out_height * CONFIG_T::out_width]
110116
) {
117+
#pragma HLS PIPELINE
118+
111119
for (int i = 0; i < CONFIG_T::pad_top; i++) {
112120
for (int j = 0; j < CONFIG_T::out_width; j++) {
113121
for(int k = 0; k < CONFIG_T::n_chan; k++) {

0 commit comments

Comments
 (0)