Skip to content

Commit

Permalink
Update on "Traceable wrapper subclass support for deferred runtime as…
Browse files Browse the repository at this point in the history
…serts"


The padded dense -> jagged conversion op has the signature:
```
_fbgemm_dense_to_jagged_forward(Tensor dense, Tensor[] offsets, SymInt? total_L=None) -> Tensor
```

when `total_L` is not specified, the meta registration has a data-dependent output shape (based on `offsets[0][-1]`). Returning an unbacked SymInt here should work in theory, but traceable wrapper subclass support is missing in later code to handle deferred runtime asserts. This PR fixes this.

[ghstack-poisoned]
  • Loading branch information
jbschlosser committed May 20, 2024
2 parents 0e11f1e + eef7472 commit 471a2bc
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions docs/source/fx.experimental.rst
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ torch.fx.experimental.symbolic_shapes
CallMethodKey
PropagateUnbackedSymInts
DivideByKey
InnerTensorKey

hint_int
is_concrete_int
Expand Down

0 comments on commit 471a2bc

Please sign in to comment.