sampling_kernels: add missing device arg

Author: victorpoughon

src/torchlensmaker/sampling/sampling_kernels.py

@@ -170,7 +170,7 @@ def example_inputs(
     def example_params(
         dtype: torch.dtype, device: torch.device
     ) -> tuple[torch.Tensor, ...]:
-        return (torch.tensor(10, dtype=torch.int64),)
+        return (torch.tensor(10, dtype=torch.int64, device=device),)
 
 
 class LinspaceSampling2DKernel(FunctionalKernel):
@@ -206,8 +206,8 @@ def example_params(
         dtype: torch.dtype, device: torch.device
     ) -> tuple[torch.Tensor, ...]:
         return (
-            torch.tensor(10, dtype=torch.int64),
-            torch.tensor(11, dtype=torch.int64),
+            torch.tensor(10, dtype=torch.int64, device=device),
+            torch.tensor(11, dtype=torch.int64, device=device),
         )
 
 
@@ -240,6 +240,6 @@ def example_params(
         dtype: torch.dtype, device: torch.device
     ) -> tuple[torch.Tensor, ...]:
         return (
-            torch.tensor(10, dtype=torch.int64),
-            torch.tensor(11, dtype=torch.int64),
+            torch.tensor(10, dtype=torch.int64, device=device),
+            torch.tensor(11, dtype=torch.int64, device=device),
         )