Skip to content

viable/strict/1780267061

viable/strict/1780267061: [DyanmicSpec] Add STATIC sentinel for shapes_spec dim slots (#185165)
@laithsakka laithsakka tagged this 31 May 18:35 
## What

Adds `STATIC` as a readable alias for `None` in shapes_spec dim / scalar-arg
slots. Both are now equivalent — `STATIC` reads more clearly at the call
site.

```python
from torch.fx.experimental.dynamic_spec import ShapeVar, TensorSpec, STATIC

torch.compile(
    fn,
    shapes_spec={
        "x": TensorSpec([ShapeVar("batch"), STATIC]),   # dim 0 dynamic, dim 1 static
        "n": STATIC,                                     # scalar arg specialized
    },
)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/185165
Approved by: https://github.com/aorenste
ghstack dependencies: #184129, #184853, #185154, #185161
Assets 2
Loading