remove guard_size_oblivious from unbind

laithsakka · laithsakka · commit b8544b036399 · 2025-03-07T22:52:04.000-08:00
ghstack-source-id: d88a7cb Pull Request resolved: #148815
diff --git a/torch/_refs/__init__.py b/torch/_refs/__init__.py
@@ -3988,14 +3988,15 @@ def unflatten(a: TensorLikeType, dim: int, sizes: ShapeType) -> TensorLikeType:
 
 @register_decomposition(aten.unbind)
 def unbind(t: TensorLikeType, dim: int = 0) -> TensorSequenceType:
-    from torch.fx.experimental.symbolic_shapes import guard_size_oblivious
+    from torch.fx.experimental.symbolic_shapes import guard_or_false
 
     dim = utils.canonicalize_dim(t.ndim, dim)
     torch._check_index(
         len(t.shape) > 0,
         lambda: "Dimension specified as 0 but tensor has no dimensions",
     )
-    if guard_size_oblivious(t.shape[dim] == 0):
+
+    if guard_or_false(t.shape[dim] == 0):
         return ()
     else:
         return tuple(
