Skip to content

Commit 31af0cc

Browse files
fix output of 1d transpose parallel to be faster
1 parent b74b210 commit 31af0cc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hls4ml/templates/vivado/nnet_utils/nnet_conv1dtranspose_resource.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,8 +150,8 @@ void conv_1d_transpose_resource_cl(
150150
for (unsigned i_res = 0; i_res < mult_n_out; i_res++) {
151151
#pragma HLS UNROLL
152152

153-
// res[(output_index-CONFIG_T::pad_left)*CONFIG_T::n_filt + i_res]
154-
*(res++) = cast<data_T, res_T, typename CONFIG_T::mult_config>(acc[i_pxl][i_res][i_sw]);
153+
res[(output_index-CONFIG_T::pad_left)*mult_n_out + i_res] =
154+
cast<data_T, res_T, typename CONFIG_T::mult_config>(acc[i_pxl][i_res][i_sw]);
155155
}
156156
}
157157
}

0 commit comments

Comments
 (0)