pytorch/torch/_prims_common
Peter Bell ad76a4e1e7 [inductor] Allow sympy expressions to participate in type promotion (#115676)
In the test example we have `add(i64[10], sympy.Expr)` where
`sympy.Expr` is not considered a promoting arg so isn't factored into
the type promotion. However, in eager it would promote to float32.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/115676
Approved by: https://github.com/lezcano
ghstack dependencies: #115677, #115699, #115700
2023-12-13 22:22:37 +00:00
..
__init__.py [inductor] Allow sympy expressions to participate in type promotion (#115676) 2023-12-13 22:22:37 +00:00
wrappers.py [torch] Skip the assertion on the return type when the annotation is a forward reference (#112870) 2023-11-04 00:22:13 +00:00