Merge pull request #1976 from cornellius-gp/fix_rmatmul_test_tol

Increase tolerance on _test_rmatmul of TestLazyEvaluatedKernelTensorBatch

Author: Balandat

test/lazy/test_lazy_evaluated_kernel_tensor.py

@@ -56,7 +56,10 @@ def _test_rmatmul(self, lhs):
         for param, param_copy in zip(lazy_tensor.kernel.parameters(), lazy_tensor_copy.kernel.parameters()):
             self.assertAllClose(param.grad, param_copy.grad, rtol=1e-3)
         self.assertAllClose(
-            lazy_tensor.x1.grad + lazy_tensor.x2.grad, lazy_tensor_copy.x1.grad + lazy_tensor_copy.x2.grad, rtol=1e-3
+            lazy_tensor.x1.grad + lazy_tensor.x2.grad,
+            lazy_tensor_copy.x1.grad + lazy_tensor_copy.x2.grad,
+            rtol=1e-3,
+            atol=1e-4,
         )
 
     def _test_inv_matmul(self, rhs, lhs=None, cholesky=False):