We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c330fd7 commit dc988dbCopy full SHA for dc988db
src/ATen/native/xpu/sycl/AdaptiveAveragePooling2dKernels.cpp
@@ -656,15 +656,18 @@ struct AdaptiveAvgPool2dKernelFunctor_cl {
656
auto read = input_
657
[_ic + _iw * oc_vec_ + _ih * oc_vec_ * iw_ +
658
_ib * ih_ * iw_ * oc_vec_];
659
+#pragma unroll
660
for (int v = 0; v < vec_size; v++) {
661
sum[v] += opmath_t(read[v]);
662
}
663
664
665
666
667
sum[v] /= kh * kw;
668
669
vec_t output_value;
670
671
672
output_value[v] = static_cast<scalar_t>(sum[v]);
673
0 commit comments