From 1d15866172612d043cce3640f4bffb42b21347ba Mon Sep 17 00:00:00 2001 From: Fritz Obermeyer Date: Fri, 8 Jul 2022 10:02:27 -0400 Subject: [PATCH] Allow autocorrelation() to run without mkl --- pyro/ops/stats.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/pyro/ops/stats.py b/pyro/ops/stats.py index 3f24ed2e41..6f906f43fc 100644 --- a/pyro/ops/stats.py +++ b/pyro/ops/stats.py @@ -93,11 +93,6 @@ def autocorrelation(input, dim=0): :param int dim: the dimension to calculate autocorrelation. :returns torch.Tensor: autocorrelation of ``input``. """ - if (not input.is_cuda) and (not torch.backends.mkl.is_available()): - raise NotImplementedError( - "For CPU tensor, this method is only supported " "with MKL installed." - ) - # Adapted from Stan implementation # https://github.com/stan-dev/math/blob/develop/stan/math/prim/mat/fun/autocorrelation.hpp N = input.size(dim)