def _compare_multi(n, k, p): data, means, covs = _make_test_case(n, k, p) # cpu in PyMC pyresult = compat.python_mvnpdf(data, means, covs) # gpu result = gps.mvnpdf_multi(data, means, covs) return result, pyresult
def _compare_single(n, k): data, means, covs = _make_test_case(n, k, 1) mean = means[0] cov = covs[0] # cpu in PyMC pyresult = compat.python_mvnpdf(data, [mean], [cov]).squeeze() # gpu result = gps.mvnpdf(data, mean, cov) return result, pyresult