pytorch
cff4d3bb - inductor: fix convert_shape_to_symint (#93349)

Commit
2 years ago
inductor: fix convert_shape_to_symint (#93349) Fixes https://github.com/pytorch/pytorch/issues/93833. When `lst` is composed of a mix of static shapes and `sympy.Expr`, convert static shapes to ints and `sympy.Expr` to `symints`. The old logic required that all of the elements of `lst` be static and it can then convert them to ints. Pull Request resolved: https://github.com/pytorch/pytorch/pull/93349 Approved by: https://github.com/jgong5, https://github.com/jansel
Author
Committer
Parents
Loading