fix autodetermination of plate device (#3405)

ilia-kats · web-flow · commit ff6ae1c95725 · 2024-10-24T08:11:43.000-04:00
Apparently, the torch.Tensor() constructor ignores device contexts
diff --git a/pyro/poutine/subsample_messenger.py b/pyro/poutine/subsample_messenger.py
@@ -45,7 +45,7 @@ def __init__(
                     )
                 )
         with ignore_jit_warnings(["torch.Tensor results are registered as constants"]):
-            self.device = device or torch.Tensor().device
+            self.device = device or torch.tensor(tuple()).device
 
     @ignore_jit_warnings(["Converting a tensor to a Python boolean"])
     def sample(self, sample_shape: torch.Size = torch.Size()) -> torch.Tensor:

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ def __init__(`
`45`	`45`	`)`
`46`	`46`	`)`
`47`	`47`	`with ignore_jit_warnings(["torch.Tensor results are registered as constants"]):`
`48`		`- self.device = device or torch.Tensor().device`
	`48`	`+ self.device = device or torch.tensor(tuple()).device`
`49`	`49`
`50`	`50`	`@ignore_jit_warnings(["Converting a tensor to a Python boolean"])`
`51`	`51`	`def sample(self, sample_shape: torch.Size = torch.Size()) -> torch.Tensor:`