Skip to content

Commit dc988db

Browse files
committed
unroll
1 parent c330fd7 commit dc988db

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/ATen/native/xpu/sycl/AdaptiveAveragePooling2dKernels.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -656,15 +656,18 @@ struct AdaptiveAvgPool2dKernelFunctor_cl {
656656
auto read = input_
657657
[_ic + _iw * oc_vec_ + _ih * oc_vec_ * iw_ +
658658
_ib * ih_ * iw_ * oc_vec_];
659+
#pragma unroll
659660
for (int v = 0; v < vec_size; v++) {
660661
sum[v] += opmath_t(read[v]);
661662
}
662663
}
663664
}
665+
#pragma unroll
664666
for (int v = 0; v < vec_size; v++) {
665667
sum[v] /= kh * kw;
666668
}
667669
vec_t output_value;
670+
#pragma unroll
668671
for (int v = 0; v < vec_size; v++) {
669672
output_value[v] = static_cast<scalar_t>(sum[v]);
670673
}

0 commit comments

Comments
 (0)