Fix stride in TensorMetadata to always be a Tuple[int, ...] (#141106)

marvin-kim · facebook-github-bot · commit cfe1d500593a · 2024-11-19T21:09:00.000-08:00
Summary: Pull Request resolved: #141106 Test Plan: CI Differential Revision: D66204410
diff --git a/torch/fx/passes/shape_prop.py b/torch/fx/passes/shape_prop.py
@@ -41,7 +41,7 @@ def _extract_tensor_metadata(
     shape = result.shape
     dtype = result.dtype
     requires_grad = result.requires_grad
-    stride = result.stride() if not is_sparse_any(result) else None
+    stride = result.stride() if not is_sparse_any(result) else ()
 
     memory_format = None
 
