Skip to content

Commit a965531

Browse files
authored
out of bounds (#5)
1 parent 3c6fb60 commit a965531

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/vpu/shave/VPUEM_piecewise_calc_subblk_size.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ class VPUEMCalcSubblk {
8181
}
8282
}
8383

84-
if (m == 2) {
84+
if (m == 3) {
8585
blk_tensor.set_shape(m - 3, round_dim(max_num_data,
8686
(blk_tensor.get_shape()[m - 1] * blk_tensor.get_shape()[m - 2])));
8787
if (blk_tensor.get_shape()[m - 3] > int(ts.get_shape()[m - 2])) {

0 commit comments

Comments
 (0)