amend

pytorch · Oct 10, 2023 · fb5c470 · fb5c470
1 parent 85f596d
commit fb5c470
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tensordict/memmap.py b/tensordict/memmap.py
@@ -849,8 +849,6 @@ def _where(condition, input, other):
         input = input.to("cpu").to(device, non_blocking=True)
     else:
         input = input.as_tensor()
-    if condition.device != device or (isinstance(other, (MemmapTensor, torch.Tensor)) and other.device != device):
-        raise ValueError(f"{condition.device}, {device}, {other.device}")
     return torch.where(condition=condition, input=input, other=other)