Update gpytorch/lazy/kronecker_product_added_diag_lazy_tensor.py
@@ -30,7 +30,7 @@ class KroneckerProductAddedDiagLazyTensor(AddedDiagLazyTensor):
# solve but we only want to cache the probe vectors for the backwards
with skip_logdet_forward(True):
inv_quad_term, func_logdet_term = super().inv_quad_logdet(
- inv_quad_rhs=inv_quad_rhs, logdet=logdet is not None, reduce_inv_quad=reduce_inv_quad
+ inv_quad_rhs=inv_quad_rhs, logdet=logdet, reduce_inv_quad=reduce_inv_quad
)
if logdet is not None:
