pytorch

7d7a3f0b - [inductor] Support sympy.expr in user-defined Triton kernel grid fn (#119165)

[inductor] Support sympy.expr in user-defined Triton kernel grid fn (#119165) ## Problem A user-defined Triton kernel grid may use a sympy magic method like `Max`. This comes in the form of a form of a `sympy.Expr`, namely `sympy.core.function.FunctionClass`. Handling this is not trivial since `user_defined_kernel_grid_fn_code` is used in Eager & Inductor. Eager usage below. ## Approach Pass in wrapper when Inductor codegens grid with ints/sympy.Expr, so we can utilize wrapper functions, such as `codegen_shape_tuple()`. Differential Revision: D53367012 Pull Request resolved: https://github.com/pytorch/pytorch/pull/119165 Approved by: https://github.com/aakhundov