hotfix: disable use of QR for preconditioning due to speed regression in pytorch.
@@ -63,7 +63,8 @@ class AddedDiagLazyTensor(SumLazyTensor):
)
return None, None, None
- if self._piv_chol_self.dim() == 2: # TODO: Whenever PyTorch supports batch mode
+ # TODO: Hotfix removing use of QR due to drastic speed regression in PyTorch
+ if False and self._piv_chol_self.dim() == 2: # TODO: Whenever PyTorch supports batch mode
*batch_shape, n, k = self._piv_chol_self.shape
self._noise = self._diag_tensor.diag().unsqueeze(-1)
