pytorch
1faef895 - Inductor cpp wrapper: support sympy.Expr as input (#101257)

Commit
1 year ago
Inductor cpp wrapper: support sympy.Expr as input (#101257) Leverage the logic in https://github.com/pytorch/pytorch/pull/95533 to get the `dtype` of `sympy.Expr` and support it as graph input in the cpp wrapper. Pull Request resolved: https://github.com/pytorch/pytorch/pull/101257 Approved by: https://github.com/jgong5, https://github.com/Skylion007, https://github.com/EikanWang, https://github.com/jansel
Author
Committer
Parents
Loading