Skip to content

Commit 607415e

Browse files
Avoid unnecessary memory allocation for covariance downdate in SGPR prediction strategy (#2559)
* Added ability for priors of transformed distributions to have their p… (#2551) (#2552) * avoid unnecessary memory allocation for downdate in SGPR prediction --------- Co-authored-by: David Eriksson <[email protected]>
1 parent 917603c commit 607415e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gpytorch/models/exact_prediction_strategies.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -865,5 +865,5 @@ def exact_predictive_covar(self, test_test_covar, test_train_covar):
865865
"This is likely a bug in GPyTorch."
866866
)
867867

868-
res = test_test_covar - (L @ (covar_cache @ L.transpose(-1, -2)))
868+
res = test_test_covar - MatmulLinearOperator(L, covar_cache @ L.mT)
869869
return res

0 commit comments

Comments
 (0)