Skip flaky test_autocast_cpu test (#8580)

Author: NicolasHug

test/test_ops.py

@@ -507,11 +507,13 @@ def test_autocast(self, aligned, deterministic, x_dtype, rois_dtype):
  rois_dtype=rois_dtype,
  )
 
+ @pytest.mark.skip(reason="1/5000 flaky failure")
  @pytest.mark.parametrize("aligned", (True, False))
  @pytest.mark.parametrize("deterministic", (True, False))
  @pytest.mark.parametrize("x_dtype", (torch.float, torch.bfloat16))
  @pytest.mark.parametrize("rois_dtype", (torch.float, torch.bfloat16))
  def test_autocast_cpu(self, aligned, deterministic, x_dtype, rois_dtype):
+
  with torch.cpu.amp.autocast():
  self.test_forward(
  torch.device("cpu"),