Skip to content

Commit

Permalink
Latest GPyTorch
Browse files Browse the repository at this point in the history
Summary:
Commit hash: c62c324aff0732ab0a1b6f37119e27f4612e33a7

Generated using `./pytorch/fb_build/import_gpytorch.sh`

This includes removal of deprecated lazy tensor (cornellius-gp/gpytorch#2615), which eliminates a significant overhead.

Reviewed By: Balandat

Differential Revision: D67141046

fbshipit-source-id: 34ea7a8c694a51d02e93512c2e5c08e532f0805b
  • Loading branch information
saitcakmak authored and facebook-github-bot committed Dec 13, 2024
1 parent 28243d7 commit 1144806
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_pairwise_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def test_kernelgrad_pairwise(self):
c = torch.cat((x2[..., :2], deriv_idx_2), dim=1)
d = torch.cat((x2[..., 2:-1], deriv_idx_2), dim=1)

c12 = kernel.forward(x1, x2).evaluate().detach().numpy()
c12 = kernel.forward(x1, x2).to_dense().detach().numpy()
pwc = (
(
kernel.latent_kernel.forward(a, c)
Expand Down Expand Up @@ -73,7 +73,7 @@ def test_covar(self):
b = x1[..., 2:]
c = x2[..., :2]
d = x2[..., 2:]
c12 = self.kernel.forward(x1, x2).evaluate().detach().numpy()
c12 = self.kernel.forward(x1, x2).to_dense().detach().numpy()
pwc = (
(
self.latent_kernel.forward(a, c)
Expand All @@ -95,7 +95,7 @@ def test_covar(self):
b = x3[..., 2:]
c = x4[..., :2]
d = x4[..., 2:]
c34 = self.kernel.forward(x3, x4).evaluate().detach().numpy()
c34 = self.kernel.forward(x3, x4).to_dense().detach().numpy()
pwc = (
(
self.latent_kernel.forward(a, c)
Expand All @@ -122,7 +122,7 @@ def test_latent_diag(self):

# should get 0 variance on pairs (a,a)
diag = torch.cat((a, a), dim=1)
diagv = self.kernel.forward(diag, diag).evaluate().detach().numpy()
diagv = self.kernel.forward(diag, diag).to_dense().detach().numpy()
npt.assert_allclose(diagv, 0.0)

def test_diag(self):
Expand Down

0 comments on commit 1144806

Please sign in to comment.