Fix test_avg_pool3d issue in pytorch_paralleltbb_linux_xenial_py3_6_gcc5_4_test (#36103)

Di Wu · facebook-github-bot · commit 16d9bcd725e9 · 2020-04-07T08:26:20.000-07:00
Summary: Fix parallel execution issue introduced by #35740 Pull Request resolved: #36103 Test Plan: test_quantized.py Differential Revision: D20879323 Pulled By: allwu fbshipit-source-id: a2deaaf5c933cbef3096a399c19c44d28935bd69
diff --git a/aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp b/aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp
@@ -856,8 +856,8 @@ void do_avg_pool_nhwc_on_AVX2(
   constexpr int cb_size = 16;
   constexpr int vec_width = Vec256<T>::size() / 4;
   constexpr int cb_step = cb_size * vec_width;
-  static Vec256<int32_t> acc_buffer[cb_size];
-  static Vec256<float> acc_buffer_fp[cb_size];
+  Vec256<int32_t> acc_buffer[cb_size];
+  Vec256<float> acc_buffer_fp[cb_size];
 
   if (vec_width == 8) {
     for (int c = c_start; c < csize; c += cb_step) {
diff --git a/test/quantization/test_quantized.py b/test/quantization/test_quantized.py
@@ -890,7 +890,6 @@ def test_avg_pool2d_nhwc(self, X, kernel, stride, padding, ceil_mode, count_incl
                              message=error_message.format(name + '.zero_point', scale,
                              X_hat.q_zero_point()))
 
-    @unittest.skip("Fix me! to stop breaking the builds")
     @given(X=hu.tensor(shapes=hu.array_shapes(min_dims=5, max_dims=5,
                                               min_side=5, max_side=10),
                        qparams=hu.qparams(dtypes=torch.quint8)),
