[tensorexpr] Add 'pre_alloc' argument in python API of tensorexpr kernel (#64718)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64718
Test Plan: Imported from OSS
Reviewed By: navahgar
Differential Revision: D30826582
Pulled By: huiguoo
fbshipit-source-id: 6c173c8964f2643039273cdc83e64fb02bb5f381